如何使用每个唯一过滤器编写多个CSV

import csv

def write_csv_files(source_file, unique_filter_field):
    # 读取CSV文件
    with open(source_file, 'r') as csvfile:
        reader = csv.DictReader(csvfile)
        data = list(reader)
    
    # 创建唯一过滤器
    unique_values = set()
    for row in data:
        unique_values.add(row[unique_filter_field])
    
    # 创建多个CSV文件
    for value in unique_values:
        filtered_data = [row for row in data if row[unique_filter_field] == value]
        file_name = f"{value}.csv"  # 基于唯一值创建文件名
        with open(file_name, 'w', newline='') as csvfile:
            writer = csv.DictWriter(csvfile, fieldnames=reader.fieldnames)
            writer.writeheader()
            writer.writerows(filtered_data)

# 示例调用
write_csv_files('source.csv', 'filter_field')