从文件夹中打开多个CSVs按列中的值拆分它们

import os
import csv

# 指定文件夹路径
folder_path = "path/to/folder"

# 获取文件夹中的所有CSV文件
csv_files = [file for file in os.listdir(folder_path) if file.endswith(".csv")]

# 遍历每个CSV文件
for file in csv_files:
    file_path = os.path.join(folder_path, file)
    
    # 打开CSV文件
    with open(file_path, "r") as csv_file:
        reader = csv.reader(csv_file)
        
        # 创建字典用于存储拆分后的数据
        data_dict = {}
        
        # 遍历每一行数据
        for row in reader:
            # 获取拆分列的值
            column_value = row[column_index]
            
            # 检查字典中是否已存在该列值对应的键
            if column_value in data_dict:
                # 如果存在，则将当前行添加到对应的值（列表）中
                data_dict[column_value].append(row)
            else:
                # 如果不存在，则创建新的键值对，并将当前行作为初始值
                data_dict[column_value] = [row]
        
        # 遍历拆分后的数据字典
        for key, value in data_dict.items():
            # 创建新的CSV文件名
            new_file_name = f"{file}_{key}.csv"
            new_file_path = os.path.join(folder_path, new_file_name)
            
            # 将拆分后的数据写入新的CSV文件
            with open(new_file_path, "w", newline="") as new_csv_file:
                writer = csv.writer(new_csv_file)
                writer.writerows(value)