Python Pandas按SalesContact导出多Excel文件异常排查与修复

常见错误代码示例 #

以下是导致所有数据写入单个文件的典型错误实现：

import pandas as pd

# 读取原始销售数据
df = pd.read_excel("sales_data.xlsx")

# 错误实现：循环外创建单个ExcelWriter，所有分组写入同一文件
writer = pd.ExcelWriter("quotation.xlsx", engine="openpyxl")
for contact, group in df.groupby("SalesContact"):
    group.to_excel(writer, sheet_name=contact, index=False)
writer.save()

问题根源 #

• 在循环外部初始化了单个ExcelWriter实例，所有SalesContact的分组数据都被写入同一个文件的不同工作表，而非独立文件。
• 未针对每个SalesContact生成唯一的输出文件名，导致数据无法按联系人拆分。

修复方案 #

修复后的代码会为每个SalesContact生成独立的Excel文件：

import pandas as pd
import os

# 创建专属输出目录（自动跳过已存在的目录）
output_dir = "contact_quotations"
os.makedirs(output_dir, exist_ok=True)

# 读取原始销售数据
df = pd.read_excel("sales_data.xlsx")

# 按SalesContact分组并逐个导出
for contact, group in df.groupby("SalesContact"):
    # 生成无特殊字符的唯一文件名
    clean_contact_name = contact.replace(" ", "_").replace("/", "-")
    output_file = os.path.join(output_dir, f"quotation_{clean_contact_name}.xlsx")
    
    # 为每个联系人创建独立的ExcelWriter，使用with语句自动管理资源
    with pd.ExcelWriter(output_file, engine="openpyxl") as writer:
        group.to_excel(writer, sheet_name="Quotation_Details", index=False)
    
    print(f"成功生成报价文件：{output_file}")

修复关键点 #

• 循环内创建独立ExcelWriter：每次迭代都初始化新的ExcelWriter实例，确保每个分组写入单独文件。
• 生成唯一文件名：对SalesContact名称做字符清理（替换空格、斜杠等），避免文件路径错误。
• 自动管理资源：使用with语句自动完成文件的保存与关闭，无需手动调用save()或close()。
• 创建输出目录：提前创建专门的输出文件夹，避免文件散落在根目录，同时防止路径不存在的报错。

内容的提问来源于stack exchange，提问作者Burritodeltodo