c

2026-06-05 21:11:32 +08:00
parent d8cec42fa9
commit 9ba8306b8d
6 changed files with 550 additions and 144 deletions
@@ -1,63 +0,0 @@
-import os
-import pandas as pd
-from pathlib import Path
-import sys
-
-def merge_transaction_data():
-    # 获取用户输入的路径
-    input_path_str = input("请输入包含 Excel 文件的文件夹路径: ").strip()
-    input_dir = Path(input_path_str)
-
-    # 检查路径是否有效
-    if not input_dir.is_dir():
-        print(f"错误：提供的路径 '{input_path_str}' 不是一个有效的目录。")
-        return
-
-    all_data = []
-    target_sheet = "交易流水"
-    
-    # 获取目录下所有的 excel 文件 (支持 .xlsx 和 .xls)
-    files = list(input_dir.glob("*.xlsx")) + list(input_dir.glob("*.xls"))
-    
-    if not files:
-        print("在该目录下未找到任何 Excel 文件。")
-        return
-
-    for file_path in files:
-        try:
-            # 读取指定的 sheet
-            # 使用 pandas 的 read_excel，如果 sheet 不存在会抛出 ValueError
-            df = pd.read_excel(file_path, sheet_name=target_sheet)
-            
-            if df.empty:
-                print(f"跳过 {file_path.name}：工作表 '{target_sheet}' 为空。")
-                continue
-                
-            all_data.append(df)
-            print(f"成功读取: {file_path.name}")
-        except ValueError:
-            # 如果找不到指定的 sheet 名称，则跳过该文件
-            print(f"跳过 {file_path.name}：未找到名为 '{target_sheet}' 的工作表。")
-        except Exception as e:
-            print(f"处理文件 {file_path.name} 时发生错误: {e}")
-
-    if all_data:
-        # 合并所有 DataFrame
-        merged_df = pd.concat(all_data, ignore_index=True)
-        
-        # 定义输出路径
-        output_file = input_dir / "txn_summary.xlsx"
-        
-        # 保存结果
-        # 使用 openpyxl 作为引擎以确保兼容性
-        merged_df.to_excel(output_file, index=False, engine='openpyxl')
-        print("-" * 30)
-        print(f"合并成功！")
-        print(f"总计处理文件数: {len(all_data)}")
-        print(f"结果已保存至: {output_file}")
-    else:
-        print("未找到包含 '" + target_sheet + "' 工作表的有效数据，未生成输出文件。")
-
-if __name__ == "__main__":
-    # 确保安装了必要的库：pip install pandas openpyxl
-    merge_transaction_data()