当多个文件合并在一起时，如何将文件名添加到数据框中的列？

import pandas as pd
import os

folder_path = '路径/to/文件夹'  # 替换为你的文件夹路径
file_extension = '.csv'  # 替换为你的文件扩展名

file_names = []  # 存储文件名的列表
data_frames = []  # 存储数据框的列表

# 遍历文件夹中的文件
for file in os.listdir(folder_path):
    if file.endswith(file_extension):
        file_names.append(file.split('.')[0])  # 获取文件名并添加到列表
        
        file_path = os.path.join(folder_path, file)  # 获取文件路径
        df = pd.read_csv(file_path)  # 读取CSV文件
        data_frames.append(df)  # 将数据框添加到列表
        
# 合并数据框
merged_df = pd.concat(data_frames)

# 创建新列并将文件名赋值给该列
merged_df['文件名'] = file_names

# 打印合并后的数据框
print(merged_df)