如何根据云数据流python sdk中的列值拆分csv文件

import csv
from qcloud_cos import CosS3Client

secret_id = 'your-secret-id'
secret_key = 'your-secret-key'
region = 'your-region'
bucket = 'your-bucket'

cos_client = CosS3Client(secret_id, secret_key, region)

def split_csv_file(file_key, column_name):
    # 下载 CSV 文件
    local_file_path = '/path/to/local/file.csv'
    cos_client.download_file(bucket, file_key, local_file_path)
    
    # 按列值拆分 CSV 文件
    with open(local_file_path, 'r') as input_file:
        csv_reader = csv.DictReader(input_file)
        csv_data = {}
        
        for row in csv_reader:
            column_value = row[column_name]
            if column_value in csv_data:
                csv_data[column_value].append(row)
            else:
                csv_data[column_value] = [row]
    
    # 保存拆分后的 CSV 文件
    for column_value, rows in csv_data.items():
        output_file_path = f'/path/to/output/{column_value}.csv'
        with open(output_file_path, 'w') as output_file:
            csv_writer = csv.DictWriter(output_file, fieldnames=csv_reader.fieldnames)
            csv_writer.writeheader()
            csv_writer.writerows(rows)
        
        # 上传拆分后的 CSV 文件到 COS
        cos_client.upload_file(bucket, output_file_path, f'output/{column_value}.csv')

file_key = 'your-csv-file-key.csv'
column_name = 'your-column-name'
split_csv_file(file_key, column_name)