在一个csv文件中查找另一个csv文件的值，如果找到值，则将这些值写入另一个csv中

import csv

def find_and_write_csv(file1, file2, output_file):
    # 读取第一个csv文件
    with open(file1, 'r') as f1:
        csv_reader1 = csv.reader(f1)
        data1 = list(csv_reader1)

    # 读取第二个csv文件
    with open(file2, 'r') as f2:
        csv_reader2 = csv.reader(f2)
        data2 = list(csv_reader2)

    # 查找匹配值并写入第三个csv文件
    with open(output_file, 'w', newline='') as f3:
        csv_writer = csv.writer(f3)
        for row1 in data1:
            for row2 in data2:
                if row1[0] == row2[0]:  # 假设比较第一列的值
                    csv_writer.writerow(row1)

# 调用函数进行查找和写入
find_and_write_csv('file1.csv', 'file2.csv', 'output.csv')