读取多个CSV并根据文件名python中的年份将这些CSV合并到单独的文件中

import pandas as pd

import os

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

# 创建一个字典来存储按年份分类的CSV数据
csv_data = {}

# 遍历所有CSV文件
for file in csv_files:
    # 提取文件名中的年份信息
    year = file.split('.')[0][-4:]
    
    # 读取CSV文件数据
    data = pd.read_csv(file)
    
    # 将数据存储到字典中对应的年份键下
    if year in csv_data:
        csv_data[year].append(data)
    else:
        csv_data[year] = [data]

# 遍历按年份分类的CSV数据
for year, data_list in csv_data.items():
    # 合并数据
    merged_data = pd.concat(data_list)
    
    # 将合并后的数据保存到单独的文件中
    merged_data.to_csv(f'{year}_merged.csv', index=False)