使用Python查找两个CSV文件中的差异

import csv
from difflib import SequenceMatcher

def compare_csv(file1, file2):
    # 打开第一个CSV文件并读取为列表或字典
    with open(file1, 'r') as f1:
        reader1 = csv.reader(f1)
        data1 = list(reader1)

    # 打开第二个CSV文件并读取为列表或字典
    with open(file2, 'r') as f2:
        reader2 = csv.reader(f2)
        data2 = list(reader2)

    # 使用SequenceMatcher类比较两个列表或字典的差异
    matcher = SequenceMatcher(None, data1, data2)
    differences = matcher.get_opcodes()

    # 输出差异的结果
    for tag, i1, i2, j1, j2 in differences:
        if tag == 'insert':
            print("在第二个CSV文件中插入了以下行：")
            for line in data2[j1:j2]:
                print(line)
        elif tag == 'delete':
            print("在第一个CSV文件中删除了以下行：")
            for line in data1[i1:i2]:
                print(line)
        elif tag == 'replace':
            print("在第一个CSV文件中替换了以下行：")
            for line in data1[i1:i2]:
                print(line)
            print("替换为以下行：")
            for line in data2[j1:j2]:
                print(line)

    # 将差异的结果输出到一个新的CSV文件中
    with open('diff.csv', 'w', newline='') as f:
        writer = csv.writer(f)
        for tag, i1, i2, j1, j2 in differences:
            if tag != 'equal':
                writer.writerow([tag] + data1[i1:i2][0] + data2[j1:j2][0])

# 调用函数并传入两个CSV文件的路径
compare_csv('file1.csv', 'file2.csv')