Question

我正在尝试使用存储在几十行的csv文件中的关键字来搜索行数超过150K的csv文件。最好的方法是什么？我已经尝试了几件事，但是什么都没有使我走得很远。

当前代码：

import csv
import pandas as pd
data = pd.read_csv('mycsv.csv')
for line in data:
    if 'Apple' in line:
        print(line)

这不是我想要的，而是我目前拥有的。 for循环是我尝试使用较小的csv文件中的关键字之一来获取输出的尝试。到目前为止，我要么出错，要么没有输出。

编辑：我忘了提到我要搜索的较大的csv文件来自Web链接，因此我认为open不起作用

Answer 1

假设您的关键字存储在名为keys.csv的文件中，并且在该文件的每一行中，只有一个关键字，如下所示：

Orange
Apple
...

然后尝试一下：

with open('mycsv.csv') as mycsv, open('keys.csv') as keys:
    keys = keys.readlines()
    for line in mycsv:
        if any([key[:-2] in line for key in keys]):
            print(line)

如何在csv文件中搜索存储在另一个csv文件中的关键字？

1 个答案: