import os
import pandas as pd


def process_csv_files(folder_path):
    all_data = {}

    # 遍历文件夹及子文件夹中的所有csv文件
    for root, dirs, files in os.walk(folder_path):
        for filename in files:
            if filename.endswith(".csv"):
                file_path = os.path.join(root, filename)

                # 提取姓名和表格类型
                name, table_type = os.path.splitext(filename)[0].split('-')

                # 读取CSV文件
                df = pd.read_csv(file_path)

                # 初始化字典结构
                if table_type not in all_data:
                    all_data[table_type] = {}

                if name not in all_data[table_type]:
                    all_data[table_type][name] = {}

                # 遍历每一行，记录题目编号和对应的选项
                for _, row in df.iterrows():
                    question_number = row['题目编号']
                    answer = row['选项']
                    all_data[table_type][name][question_number] = answer

    # 将数据转换为DataFrame并保存
    for table_type, data in all_data.items():
        combined_df = pd.DataFrame(data).T  # 转置表格，使姓名为行，题目编号为列
        combined_df.to_csv(f"{table_type}.csv", encoding='utf-8-sig')
        print(f"保存 {table_type}.csv")


# 示例文件夹路径
folder_path = r'C:\Users\Zherui\Desktop\Project\7\答卷'  # 替换为你的文件夹路径
process_csv_files(folder_path)