使用pandas提取含有关键词的行
可以使用pandas的str.contains()方法来提取含有关键词的行,具体步骤如下:
- 导入pandas库并读取数据文件:
import pandas as pd
df = pd.read_csv('data.csv')
- 使用str.contains()方法来提取含有关键词的行:
keyword = "apple"
new_df = df[df['column_name'].str.contains(keyword)]
其中,'column_name'是数据框中的列名,可以根据实际情况进行修改。'new_df'是符合条件的行组成的新数据框。
- 可以将新数据框保存到文件中:
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)
``
原文地址: https://www.cveoy.top/t/topic/cG1S 著作权归作者所有。请勿转载和采集!