Pandas 提取含有关键词的行数据 - 使用 str.contains() 方法

使用 Pandas 提取含有关键词的行数据

使用 Pandas 库中的 str.contains() 方法可以方便地提取数据框中含有特定关键词的行。以下步骤将详细介绍如何实现这一功能：

1. 导入 Pandas 库并读取数据文件：

import pandas as pd

df = pd.read_csv('data.csv')

2. 使用 str.contains() 方法提取含有关键词的行：

keyword = 'apple'
new_df = df[df['column_name'].str.contains(keyword)]

其中，'column_name' 是数据框中的列名，可以根据实际情况进行修改。'new_df' 是符合条件的行组成的新数据框。

3. 将新数据框保存到文件中：

new_df.to_csv('new_data.csv', index=False)

完整代码示例：

import pandas as pd

df = pd.read_csv('data.csv')

keyword = 'apple'
new_df = df[df['column_name'].str.contains(keyword)]

new_df.to_csv('new_data.csv', index=False)

通过以上步骤，你可以轻松提取数据框中含有特定关键词的行数据。