导入多个csv文件并向每个文件添加年份

import os
import csv

# 获取所有CSV文件的文件名
csv_files = [file for file in os.listdir('.') if file.endswith('.csv')]

# 遍历每个CSV文件
for file in csv_files:
    # 生成新的文件名，添加年份
    new_file = f"{file[:-4]}_with_year.csv"
    
    # 打开原始CSV文件和新文件
    with open(file, 'r') as csv_file, open(new_file, 'w', newline='') as new_csv_file:
        reader = csv.reader(csv_file)
        writer = csv.writer(new_csv_file)
        
        # 读取原始CSV文件的每一行数据
        for row in reader:
            # 在每一行数据的末尾添加年份
            row.append('2022')  # 这里的年份可以根据实际需求进行修改
            
            # 将修改后的行写入新文件
            writer.writerow(row)