按列名合并两个cvs文件

import csv

def merge_csv(file1, file2, merge_column):
    # 读取两个CSV文件
    with open(file1, 'r') as f1, open(file2, 'r') as f2:
        reader1 = csv.DictReader(f1)
        reader2 = csv.DictReader(f2)
        
        # 解析CSV文件
        data1 = list(reader1)
        data2 = list(reader2)
        
        # 确定合并列名
        merge_column = merge_column.lower()
        
        # 合并数据
        merged_data = []
        for row1 in data1:
            for row2 in data2:
                if row1[merge_column] == row2[merge_column]:
                    merged_row = {**row1, **row2}
                    merged_data.append(merged_row)
        
        # 写入合并后的数据
        with open('merged.csv', 'w', newline='') as f:
            writer = csv.DictWriter(f, fieldnames=merged_data[0].keys())
            writer.writeheader()
            writer.writerows(merged_data)

# 调用合并函数
merge_csv('file1.csv', 'file2.csv', 'column_name')