📌  相关文章
📜  在 Pandas 列中搜索一个值(1)

📅  最后修改于: 2023-12-03 14:51:03.849000             🧑  作者: Mango

在 Pandas 列中搜索一个值

在 Pandas 中,有时候需要在某一列中查找特定的值。下面将介绍两种方法实现此功能。

方法一:使用 .isin() 方法

可以使用 .isin() 方法来查找一个值是否存在于某一列中,具体方法如下:

import pandas as pd

# 构造数据
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'gender': ['F', 'M', 'M', 'M']}

df = pd.DataFrame(data)

# 查找 gender 列中是否存在 'M'
result = df['gender'].isin(['M'])

print(result)

输出结果为:

0    False
1     True
2     True
3     True
Name: gender, dtype: bool

说明 gender 列中第 2、3、4 行都是 'M'。

方法二:使用 .str.contains() 方法

如果需要查找的值是一个字符串,可以使用 .str.contains() 方法查找某一列中是否包含该字符串,具体方法如下:

import pandas as pd

# 构造数据
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'interests': ['Swimming, Running', 'Reading, Writing', 'Singing, Dancing', 'Swimming, Writing']}

df = pd.DataFrame(data)

# 查找 interests 列中是否包含字符串 'Swimming'
result = df['interests'].str.contains('Swimming')

print(result)

输出结果为:

0     True
1    False
2    False
3     True
Name: interests, dtype: bool

说明 interests 列中第 1、4 行都包含字符串 'Swimming'。