【效率神器】一键搞定！用CodeBuddy实现多Excel文件智能合并

import os
import pandas as pd


def merge_excel_files(input_folder, output_file):
    """
    合并多个具有相同表头的Excel文件

    参数:
        input_folder: 包含Excel文件的文件夹路径
        output_file: 合并后的输出文件路径
    """
    # 获取所有Excel文件
    excel_files = [f for f in os.listdir(input_folder)
                   if f.endswith(('.xls', '.xlsx', '.xlsm', '.xlsb'))]

    if not excel_files:
        print("错误: 没有找到Excel文件")
        return False

    # 读取第一个文件获取表头
    first_file = os.path.join(input_folder, excel_files[0])
    try:
        header = pd.read_excel(first_file, nrows=0).columns.tolist()
        print(f"表头: {header}")
    except Exception as e:
        print(f"错误: 无法读取第一个文件的表头 - {e}")
        return False

    # 创建一个空的DataFrame用于存储合并后的数据
    merged_data = pd.DataFrame(columns=header)

    # 合并所有文件
    for file in excel_files:
        file_path = os.path.join(input_folder, file)
        try:
            # 读取文件数据(跳过第一行表头)
            df = pd.read_excel(file_path, skiprows=1, header=None)
            df.columns = header  # 设置相同的表头
            merged_data = pd.concat([merged_data, df], ignore_index=True)
            print(f"已合并: {file} (包含 {len(df)} 行数据)")
        except Exception as e:
            print(f"警告: 跳过文件 {file} - 错误: {e}")
            continue

    # 保存合并后的文件
    try:
        merged_data.to_excel(output_file, index=False)
        print(f"\n合并完成! 共合并 {len(excel_files)} 个文件")
        print(f"总行数: {len(merged_data)}")
        print(f"已保存到: {output_file}")
        return True
    except Exception as e:
        print(f"错误: 无法保存合并后的文件 - {e}")
        return False


if __name__ == "__main__":
    # 直接使用预定义的路径
    INPUT_FOLDER = "./excel"  # Excel文件所在目录
    OUTPUT_FILE = "./merged.xlsx"  # 合并后的输出文件路径

    print("=== Excel文件合并工具 ===")
    print(f"输入目录: {INPUT_FOLDER}")
    print(f"输出文件: {OUTPUT_FILE}")

    # 检查输入目录是否存在
    if not os.path.exists(INPUT_FOLDER):
        print(f"错误: 输入目录不存在 - {INPUT_FOLDER}")
        exit()

    # 执行合并
    print("\n开始合并文件...")
    success = merge_excel_files(INPUT_FOLDER, OUTPUT_FILE)

    if success:
        print("\n操作成功完成!")
    else:
        print("\n操作完成，但可能存在问题，请检查上述警告/错误信息")