合并python中具有相似值的csv行

import csv
from collections import defaultdict

def merge_csv_rows(input_file, output_file):
    merged_rows = defaultdict(list)

    with open(input_file, 'r') as file:
        reader = csv.reader(file)
        next(reader)  # 跳过标题行

        for row in reader:
            key = row[0]  # 假设相似值在第一列
            merged_rows[key].extend(row[1:])  # 合并相似值的行数据

    with open(output_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(['Key', 'Merged Values'])  # 写入标题行

        for key, values in merged_rows.items():
            writer.writerow([key, ','.join(values)])  # 将合并后的值以逗号分隔写入行

merge_csv_rows('input.csv', 'output.csv')