python 多个csv文件分类合并

可以使用 Pandas 库来处理多个 CSV 文件的分类合并。下面是一个示例代码：

import pandas as pd
import os

# 定义文件夹路径和输出文件名
folder_path = 'path/to/csv/files'
output_file = 'output.csv'

# 获取文件夹下所有 CSV 文件名
file_names = [f for f in os.listdir(folder_path) if f.endswith('.csv')]

# 读取所有 CSV 文件并合并
dfs = []
for file_name in file_names:
    file_path = os.path.join(folder_path, file_name)
    df = pd.read_csv(file_path)
    df['file_name'] = file_name  # 添加文件名列
    dfs.append(df)
merged_df = pd.concat(dfs)

# 根据文件名分类并输出
for file_name in file_names:
    output_path = os.path.join(folder_path, file_name)
    filtered_df = merged_df[merged_df['file_name'] == file_name]
    filtered_df.to_csv(output_path, index=False)

# 输出所有数据到一个文件
merged_df.to_csv(output_file, index=False)

以上代码会将指定文件夹下所有 CSV 文件合并为一个 DataFrame，并在其中添加一列记录文件名。然后，根据文件名分类并输出到对应的文件中，最后将所有数据输出到一个文件中。可以根据需要修改文件夹路径和输出文件名。