使用pandas库筛选csv文件中的包含某关键词的一列

假设我们有一个名为data.csv的文件，其中包含以下内容：

id,name,age
1,John,25
2,Alice,32
3,Mark,28
4,Susan,40

我们想要筛选包含关键词“John”的姓名列。可以使用pandas库中的read_csv()函数来读取文件，并使用str.contains()函数来筛选包含关键词的行。

以下是具体的代码实现：

import pandas as pd

# 读取csv文件
data = pd.read_csv('data.csv')

# 筛选包含关键词的行
filtered_data = data[data['name'].str.contains('John')]

# 输出结果
print(filtered_data['name'])

运行结果：

0    John
Name: name, dtype: object

这里我们使用了data['name'].str.contains('John')来筛选包含关键词“John”的行，然后将结果赋值给filtered_data。最后，我们输出filtered_data中的姓名列。