📅  最后修改于: 2023-12-03 14:51:03.849000             🧑  作者: Mango
在 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'。