使用 Python 在大型 CSV 文件中搜索关键字答案

【问题标题】：Keyword search in a large CSV file using Python使用 Python 在大型 CSV 文件中搜索关键字
【发布时间】：2018-04-05 14:02:27
【问题描述】：

我想知道是否有一种方法可以使用 Python 搜索关键字列表。我可以使用 PostgreSQL 搜索我的数据。这是我的 PostgreSQL 代码

SELECT distinct ON (id) id, year, cost, description
FROM mydata
WHERE description similar to'%((hotel)||(travel)|(taxi)|(food))%';

我不知道 Python 是否是最好的方法，但我正在复制的工作使用 Python，并希望坚持使用 Python。

我可以搜索一个关键字，但不知道如何搜索多个。

 import csv
 with open('mydata.csv', 'r') as f:
    for line in f.readlines():
        if 'food' in line:
            print(line)

我需要帮助

1) 使用多个关键字搜索

2) 将数据导出回csv的方式

【问题讨论】：

假设您有合理数量的数据要搜索，pandas 将是一种简单/有效的导入、搜索和导出方式。基础知识：pandas.pydata.org/pandas-docs/stable/10min.html。这是一个专门针对您要执行的操作的答案：stackoverflow.com/questions/38185688/…

【解决方案1】：

> import pandas as pd

使用包含的多个关键字搜索

> df = pd.read_csv('mydata.csv')
> df[df.description.str.contains('hotel|travel|taxi|food')]

导出为 csv

> df.to_csv('new.csv')

【讨论】：