使用Python在大型CSV文件中進行關鍵字搜索

Question

我想知道是否有一種使用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的方法

Answer 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')