如何使用python从CSV文件中过滤两个日期之间的行，并重定向到另一个文件？

import csv
from datetime import datetime

def filter_csv_by_date(input_file, output_file, start_date, end_date):
    with open(input_file, 'r') as file:
        reader = csv.reader(file)
        header = next(reader)  # 读取并保存头部信息

        filtered_rows = []
        for row in reader:
            date_str = row[0]  # 假设日期在第一列
            date = datetime.strptime(date_str, '%Y-%m-%d')  # 将日期字符串转换为日期对象

            if start_date <= date <= end_date:
                filtered_rows.append(row)

    with open(output_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(header)  # 写入头部信息
        writer.writerows(filtered_rows)  # 写入过滤后的行

# 示例用法
start_date = datetime(2022, 1, 1)
end_date = datetime(2022, 12, 31)
filter_csv_by_date('input.csv', 'output.csv', start_date, end_date)