如何拆分\n带条件的csv

import csv

def split_csv_with_condition(csv_file, condition_column, condition_value):
    # 读取CSV文件
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        rows = list(reader)

    # 根据条件拆分CSV数据
    split_data = {}
    for row in rows:
        condition = row[condition_column]
        if condition not in split_data:
            split_data[condition] = []
        split_data[condition].append(row)

    # 保存拆分结果
    for condition, data in split_data.items():
        output_file = f"{condition}.csv"
        with open(output_file, 'w', newline='') as file:
            writer = csv.DictWriter(file, fieldnames=reader.fieldnames)
            writer.writeheader()
            writer.writerows(data)

        print(f"拆分条件为'{condition}'的数据已保存到文件'{output_file}'")

# 示例用法
csv_file = 'data.csv'
condition_column = 'Category'
condition_value = 'A'
split_csv_with_condition(csv_file, condition_column, condition_value)