python 3基于日期列从csv中删除行(字符串格式)

import csv
from datetime import datetime

def delete_rows_by_date(csv_file, date_column, target_date):
    rows_to_keep = []
    
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        for row in reader:
            date_str = row[date_column]
            date = datetime.strptime(date_str, '%Y-%m-%d')  # 将日期字符串转换为日期对象
            if date != target_date:
                rows_to_keep.append(row)
    
    with open(csv_file, 'w', newline='') as file:
        writer = csv.DictWriter(file, fieldnames=reader.fieldnames)
        writer.writeheader()
        writer.writerows(rows_to_keep)

# 示例用法
csv_file = 'data.csv'  # 替换为你的CSV文件路径
date_column = 'date'  # 替换为日期列的名称
target_date = datetime(2022, 1, 1)  # 替换为目标日期

delete_rows_by_date(csv_file, date_column, target_date)