从 csv 中删除匹配值

Question

使用 DictReader 和 DictWriter 我需要找到 file1.csv 和 file2.csv 之间的匹配值。如果找到匹配项，请将其从 file1.csv 中删除文件1.csvUserName,LastIP,LastLogonJessica_Alba,10.10.10.11,11/14/2019Karen_Edwards,10.10.10.12,11/14/2019Tracy_Chung,10.10.10.25,11/15/2019文件2.csvDepartment,UserName,LastPasswordReset,LastIPIT,Jessica_Alba,9/14/2019,10.10.10.11Accounting,Karen_Edwards,9/14/2019,10.10.10.12通过删除匹配的用户更新 file1.csv 的两个文件比较后的预期输出UserName,LastIP,LastLogonTracy_Chung,10.10.10.25,11/15/2019但是，我的代码似乎并非如此。我究竟做错了什么？data3 = []with open("file1.csv","r") as in_file1, open("file2.csv", "r") as in_file2:    reader1 = csv.DictReader(in_file1)    reader2 = csv.DictReader(in_file2)    for row2 in reader2:        for row1 in reader1:            print(row1['UserName'])            if row2['UserName'] != row1['UserName']:                data3.append(row1)print(data3)

Answer 1

一个打开的文件句柄是一个流；只要还有更多未读的行，您就可以阅读，但是一旦您读完所有行，一旦您就处于流的末尾，进一步的读取将不会产生任何结果。

不要尝试多次循环内部文件，而是将其读入内存，然后对内存中的数据结构循环多次……或者更好的是，在内存中生成一个数据结构，让您可以直接查看用户是否存在于第二个文件中，因此您根本不必循环来搜索用户。

import csv

data3 = []

with open("file2.csv", "r") as in_file2:

    reader2 = csv.DictReader(in_file2)

    # Create a set of users

    users = {row2['UserName'] for row2 in reader2}

with open("file1.csv","r") as in_file1:

    reader1 = csv.DictReader(in_file1)

    for row1 in reader1:

        if row1['UserName'] not in users:

            data3.append(row1)

print(data3)