python:根据日期将csv拆分成多个csv

import datetime
import csv

def split_csv_by_date(csv_file, output_folder):
    # 读取原始CSV文件
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        header = next(reader)  # 读取表头

        # 创建一个空字典，用于存储按日期拆分的数据
        data = {}

        # 遍历CSV文件中的每一行数据
        for row in reader:
            date_str = row[0]  # 假设日期在第一列
            date = datetime.datetime.strptime(date_str, '%Y-%m-%d').date()  # 将日期字符串转换为日期对象

            # 如果日期不在字典中，则创建一个新的CSV文件
            if date not in data:
                output_file = f"{output_folder}/{date_str}.csv"
                with open(output_file, 'w', newline='') as output:
                    writer = csv.writer(output)
                    writer.writerow(header)  # 写入表头
                    data[date] = writer  # 将日期和写入器存储到字典中

            # 将当前行写入对应的CSV文件
            data[date].writerow(row)

csv_file = 'path/to/original.csv'
output_folder = 'path/to/output'

split_csv_by_date(csv_file, output_folder)