比较 2 个 csv 文件并检查前 2 列，如果匹配，请用户决定是否覆盖，然后继续下一行

Question

我有一个用例，其中有 2 个 CSV 文件，每个 CSV 文件中有一些行，每个文件都有三列。比较 2 个 csv 文件的前 2 列，如果匹配，则询问用户输入是否要使用第二个 csv 文件中的值覆盖第一个 csv 文件中的行，如果不中止操作。

当我第一次运行 python 代码时，它应该使用从第二个 CSV 文件到第一个 csv 文件的新值更新 csv 文件，但是对于连续运行我的 python 代码，我必须检查前 2 列是否匹配并要求用户决定他是否需要覆盖这些值，因为现在第一个 csv 文件将包含第一个 csv 文件中的行。

我的代码：

import csv
import sys

def csv_file_copy():
   csv_file = input("Enter the CSV file needs to be updated ")
   csv_file_cp = input("Enter the csv file from where the data needs to be copied ")
   csvfile = open(csv_file_cp, 'r',encoding="utf-8-sig")
   reader = csv.reader(csvfile)
   csv_file_orig = open(csv_file, 'r',encoding="utf-8-sig")
   reader2 = csv.reader(csv_file_orig)
   res = []
   for row in reader:
     print("This is row", row)
     for row2 in reader2:
       print("This is row2", row2)
       if (row2[0] == row[0] and row2[1] == row[1]):
         user_input = input("Store type and store number already exists in the csv file, continue? y/n ").lower()
         if user_input == "y":
           res.append(row)
         elif user_input == "n":
            print("Aborting operation")
            sys.exit(1)
       else:
        res.append(row2)
     res.append(row)
     continue
   print (reader)
   with open(csv_file, 'w') as csv_file1:
      writer = csv.writer(csv_file1, delimiter=',')
      for row in res:
        writer.writerow(row)

csv_file_copy()

当对相同的 2 个文件第二次执行代码时，第二个 for 循环仅运行一次，因此仅匹配一个值，但大约有 10 个匹配的值，这对我不起作用。

Answer 1

如果csv_file_orig不太大（或者您的可用内存太低），那么您可以将整个内容存储到列表中。

而不是

reader2 = csv.reader(csv_file_orig)

使用Afterwards。

csv_file_orig_lines = list(csv.reader(csv_file_orig))

您可以根据需要多次迭代 csv_file_orig_lines 列表，