几十个Excel只提取几列合并?30行代码一键搞定,再也不用一个个复制粘贴了

import osimport pandas as pddef merge_specific_columns(folder_path, columns_to_keep, output_file="合并结果.xlsx", include_subfolders=False):    """    从多个Excel文件中提取指定列并合并    columns_to_keep: 要保留的列名列表，如 ['产品名称', '销售额']    include_subfolders: 是否包含子文件夹    """    # 收集所有Excel文件    excel_files = []    if include_subfolders:        for root, dirs, files in os.walk(folder_path):            for f in files:                if f.lower().endswith(('.xlsx', '.xls')):                    excel_files.append(os.path.join(root, f))    else:        excel_files = [os.path.join(folder_path, f) for f in os.listdir(folder_path)                        if f.lower().endswith(('.xlsx', '.xls'))]    if not excel_files:        print("❌ 未找到Excel文件")        return    print(f"📂 找到 {len(excel_files)} 个Excel文件")    print(f"📋 要提取的列: {', '.join(columns_to_keep)}")    all_data = []    skipped_files = []    missing_columns = []    for file_path in excel_files:        file_name = os.path.basename(file_path)        print(f"  📄 {file_name}")        try:            df = pd.read_excel(file_path)            df_columns_lower = [col.lower() for col in df.columns]            # 查找匹配的列（不区分大小写）            matched_columns = []            for col in columns_to_keep:                col_lower = col.lower()                if col_lower in df_columns_lower:                    original_col = df.columns[df_columns_lower.index(col_lower)]                    matched_columns.append(original_col)                else:                    missing_columns.append((file_name, col))                    print(f"    ⚠️ 缺少列: {col}")            if not matched_columns:                print(f"    ⚠️ 未找到任何匹配列，跳过")                skipped_files.append(file_name)                continue            # 提取指定列            df_extracted = df[matched_columns].copy()            # 添加来源列            df_extracted.insert(0, '来源文件', file_name)            all_data.append(df_extracted)            print(f"    ✅ 已提取 {len(matched_columns)} 列，{len(df)} 行")        except Exception as e:            print(f"    ❌ 读取失败: {str(e)[:30]}")            skipped_files.append(file_name)            continue    if not all_data:        print("❌ 没有成功读取任何数据")        if missing_columns:            print("\n⚠️ 缺失列信息:")            for file_name, col in missing_columns[:10]:                print(f"  {file_name}: 缺少列「{col}」")        return    # 合并所有数据    merged_df = pd.concat(all_data, ignore_index=True)    # 保存结果    output_path = os.path.join(folder_path, output_file)    merged_df.to_excel(output_path, index=False)    print(f"\n✅ 完成！成功合并 {len(all_data)} 个文件")    print(f"📊 总行数: {len(merged_df)} 行")    print(f"📁 保存至: {output_path}")    if skipped_files:        print(f"\n⚠️ 跳过 {len(skipped_files)} 个文件（格式异常或无匹配列）")    if missing_columns:        print(f"\n⚠️ {len(missing_columns)} 个文件缺少指定列")def merge_by_keyword(folder_path, keyword, output_file="按关键词合并.xlsx"):    """    按文件名关键词筛选后合并    """    excel_files = [os.path.join(folder_path, f) for f in os.listdir(folder_path)                    if f.lower().endswith(('.xlsx', '.xls')) and keyword.lower() in f.lower()]    if not excel_files:        print(f"❌ 未找到文件名包含「{keyword}」的Excel文件")        return    print(f"📂 找到 {len(excel_files)} 个匹配文件")    all_data = []    for file_path in excel_files:        file_name = os.path.basename(file_path)        try:            df = pd.read_excel(file_path)            df.insert(0, '来源文件', file_name)            all_data.append(df)            print(f"  ✅ {file_name} ({len(df)} 行)")        except Exception as e:            print(f"  ❌ {file_name} 读取失败: {str(e)[:30]}")    if not all_data:        print("❌ 没有成功读取任何文件")        return    merged_df = pd.concat(all_data, ignore_index=True)    output_path = os.path.join(folder_path, output_file)    merged_df.to_excel(output_path, index=False)    print(f"\n✅ 完成！共合并 {len(all_data)} 个文件")    print(f"📊 总行数: {len(merged_df)} 行")    print(f"📁 保存至: {output_path}")if __name__ == "__main__":    print("=" * 50)    print("📊 批量合并Excel指定列")    print("=" * 50)    print("从多个Excel中提取指定列合并成一个表")    print()    path = input("请输入Excel文件夹路径: ").strip()    if path.startswith('"') and path.endswith('"'):        path = path[1:-1]    path = path.replace('\\', '/')    if not os.path.exists(path):        print("❌ 路径不存在")        input("按回车键退出...")        exit()    print("\n1. 按指定列合并（提取指定列）")    print("2. 按关键词筛选后合并（如只合并文件名含「销售」的表）")    choice = input("请选择（1或2，默认1）: ").strip() or "1"    if choice == "2":        keyword = input("请输入关键词（如 销售）: ").strip()        if not keyword:            print("❌ 关键词不能为空")            input("按回车键退出...")            exit()        merge_by_keyword(path, keyword)    else:        print("\n请输入要保留的列名（多个列名用英文逗号分隔）")        print("例如: 产品名称,销售额,日期")        columns_input = input("列名: ").strip()        columns_to_keep = [c.strip() for c in columns_input.split(',') if c.strip()]        if not columns_to_keep:            print("❌ 至少输入一个列名")            input("按回车键退出...")            exit()        # 询问是否包含子文件夹        sub = input("是否包含子文件夹？(y/n，默认n): ").strip().lower()        include_sub = sub == 'y'        merge_specific_columns(path, columns_to_keep, include_subfolders=include_sub)    input("\n按回车键退出...")