60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
import os
|
|
import pandas as pd
|
|
import glob
|
|
|
|
def merge_excel_files(input_dir):
|
|
# 获取绝对路径并确保它是存在的
|
|
input_dir = os.path.abspath(input_dir)
|
|
if not os.path.isdir(input_dir):
|
|
print(f"错误：路径 '{input_dir}' 不是有效的目录。")
|
|
return
|
|
|
|
# 获取目录下所有的 excel 文件
|
|
# 支持 .xlsx 和 .xls 格式
|
|
file_pattern = os.path.join(input_dir, "*.xls*")
|
|
files = glob.glob(file_pattern)
|
|
|
|
all_data = []
|
|
target_sheet = "交易流水"
|
|
|
|
for file_path in files:
|
|
# 跳过输出文件本身，防止循环读取
|
|
if "txn_summary" in os.path.basename(file_path):
|
|
continue
|
|
|
|
try:
|
|
# 加载 Excel 文件
|
|
# 使用 ExcelFile 对象可以先检查 sheet 名称是否存在
|
|
xls = pd.ExcelFile(file_path)
|
|
if target_sheet in xls.sheet_names:
|
|
df = pd.read_excel(xls, sheet_name=target_sheet)
|
|
# 记录来源文件名（可选，方便核对）
|
|
df['source_file'] = os.path.basename(file_path)
|
|
all_data.append(df)
|
|
print(f"成功读取: {os.path.basename(file_path)}")
|
|
else:
|
|
print(f"跳过文件 {os.path.basename(file_path)}: 未找到名为 '{target_sheet}' 的工作表。")
|
|
except Exception as e:
|
|
print(f"处理文件 {os.path.basename(file_path)} 时出错: {e}")
|
|
|
|
if all_data:
|
|
# 合并所有 DataFrame
|
|
merged_df = pd.concat(all_data, ignore_index=True)
|
|
|
|
# 生成输出路径
|
|
output_path = os.path.join(input_dir, "txn_summary.xlsx")
|
|
|
|
# 保存到 Excel
|
|
with pd.ExcelWriter(output_path, engine='openpyxl') as writer:
|
|
merged_df.to_excel(writer, sheet_name='Summary', index=False)
|
|
|
|
print("-" * 30)
|
|
print(f"合并完成！结果已保存至: {output_path}")
|
|
else:
|
|
print("未找到任何包含 '交易流水' 工作表的 Excel 文件。")
|
|
|
|
if __name__ == "__main__":
|
|
print("请输入包含 Excel 文件的目录路径:")
|
|
user_path = input("> ").strip()
|
|
merge_excel_files(user_path)
|