我如何读入一个文件，然后从这个文件中提取每个元素并搜索另一个文件以查看它是否包含该元素。Python。

Question

我正在尝试编写一段代码，用python打开一个CSV文件，并解析每一行和每一行中的每一个元素。然后查看每个元素是否在另一个CSV文件中，如果是，则将其写入第三个文件。这是我目前的代码，通过测试，我已经确定，我的搜索算法是什么工作不正常

import csv

def generateKnownReport(mypath, GKR):
    if GKR==True:
        report = open("KnownReport.txt", "w")
        file2=frozenset(open("file","r"))
        for row in csv.reader(open("file","r"),delimiter=','):
            for item in row:
                if item in file2:
                    ##report.write(str(row))
                    print('True')
                    break
                else:
                    print('ERROR')
        report.close() 
    else:
        report = open("KnownReport.txt", "w")
        report.write("No Known Report Generated.")
        report.close()

任何帮助都将不胜感激。谢谢

Answer 1

你的问题是if item in file2:。打开file2，但不处理它in不会为您实现搜索。在文件2中搜索item之前，您至少需要加载文件2

Answer 2

唯一合理的方法是将这两个文件读入一个列表或其他iterable中，然后逐步遍历以找出差异

如果重复项不重要，则一个集合将提供更好的性能

以下是一个开始的方法：

with open('file-one.csv') as f:
    reader = csv.reader(f, delimiter=',')  # adjust accordingly
    file_one = list(reader)
with open('file-two.csv') as f:
    reader = csv.reader(f, delimiter=',')
    file_two = list(reader)

element_to_search = 0  # 0 = first column in the row
line_pairs = zip(file_one, file_two)

with open('file-three.csv','w') as f:
    for line in line_pairs:
        if line[0][element_to_search] == line[1][element_to_search]:
            f.write('{}\n'.format(line[0]))