在python中合并具有相同类别的csv文件的行

import csv
import os

def merge_csv_rows(input_folder, output_file):
    merged_data = {}
    
    # 遍历输入文件夹中的所有CSV文件
    for filename in os.listdir(input_folder):
        if filename.endswith(".csv"):
            with open(os.path.join(input_folder, filename), 'r') as file:
                csv_reader = csv.reader(file)
                
                # 跳过CSV文件的标题行
                next(csv_reader)
                
                # 遍历CSV文件的每一行数据
                for row in csv_reader:
                    category = row[0]  # 假设类别信息在第一列
                    data = row[1:]  # 假设数据从第二列开始
                    
                    # 如果类别已存在于合并的数据中，则将当前行的数据追加到该类别下
                    if category in merged_data:
                        merged_data[category].append(data)
                    else:
                        merged_data[category] = [data]
    
    # 将合并的数据写入输出文件
    with open(output_file, 'w', newline='') as file:
        csv_writer = csv.writer(file)
        
        # 写入CSV文件的标题行
        csv_writer.writerow(['Category'] + ['Data'] * len(merged_data[next(iter(merged_data))][0]))
        
        # 写入合并的数据行
        for category, data_rows in merged_data.items():
            for data_row in data_rows:
                csv_writer.writerow([category] + data_row)

input_folder = 'path/to/input/folder'
output_file = 'path/to/output/file.csv'
merge_csv_rows(input_folder, output_file)