要按照一个列表中的元素值来过滤pandas数据框的一列,可以使用isin()
函数。以下是一个示例代码:
import pandas as pd
# 创建示例数据框
data = {'Name': ['Tom', 'Nick', 'John', 'Amy', 'Emily'],
'Age': [25, 30, 35, 28, 32],
'City': ['London', 'Paris', 'New York', 'Paris', 'London']}
df = pd.DataFrame(data)
# 创建过滤列表
filter_list = ['Paris', 'New York']
# 使用isin()函数过滤数据框的一列
filtered_df = df[df['City'].isin(filter_list)]
# 打印过滤结果
print(filtered_df)
输出结果为:
Name Age City
1 Nick 30 Paris
2 John 35 New York
3 Amy 28 Paris
在上述示例中,我们首先创建了一个示例数据框df
,它包含三列:Name、Age和City。然后,我们创建了一个过滤列表filter_list
,其中包含要过滤的城市名称。最后,我们使用isin()
函数过滤了数据框的City列,并将过滤结果赋值给变量filtered_df
。最后,我们打印了过滤结果。