使用xlsxwriter格式化多Excel工作表时遇KeyError问题求助

用xlsxwriter引擎导出多工作表并格式化时，KeyError的核心原因是你在尝试获取工作表时，该表还未被写入到ExcelWriter的工作表集合中，或者用了错误的方式引用表名。下面是Pythonic的解决思路：

核心要点 #

1. 写入DataFrame后立即获取对应的工作表对象，而不是事后统一查找
2. 格式化函数直接接收工作表对象和工作簿对象，避免依赖表名查找

完整代码示例 #

1. 定义自定义格式化函数 #

def Excel_formatting(worksheet, workbook):
    # 示例：设置表头样式
    header_format = workbook.add_format({
        'bold': True,
        'align': 'center',
        'valign': 'vcenter',
        'border': 1
    })
    # 应用表头格式到第一行
    worksheet.set_row(0, 18, header_format)
    
    # 示例：自动调整列宽（根据内容长度）
    for col_num in range(len(worksheet.row(0))):
        # 获取列内容的最大长度
        max_len = 0
        for row_num in range(worksheet.nrows):
            cell_value = worksheet.cell_value(row_num, col_num)
            max_len = max(max_len, len(str(cell_value)))
        # 设置列宽（预留2个字符的边距）
        worksheet.set_column(col_num, col_num, max_len + 2)

2. 主程序逻辑 #

import pandas as pd

# 模拟拆分后的子DataFrame字典（键为工作表名，值为对应DataFrame）
split_dfs = {
    '销售数据_北京': pd.DataFrame({'日期': ['2024-01-01', '2024-01-02'], '销售额': [12000, 15000]}),
    '销售数据_上海': pd.DataFrame({'日期': ['2024-01-01', '2024-01-02'], '销售额': [18000, 22000]}),
    '销售数据_广州': pd.DataFrame({'日期': ['2024-01-01', '2024-01-02'], '销售额': [9000, 11000]})
}

# 创建ExcelWriter上下文管理器（自动处理文件打开/关闭）
with pd.ExcelWriter('多表输出.xlsx', engine='xlsxwriter') as writer:
    workbook = writer.book
    
    # 循环写入每个子表并立即格式化
    for sheet_name, df in split_dfs.items():
        # 写入DataFrame到指定工作表
        df.to_excel(writer, sheet_name=sheet_name, index=False)
        
        # 直接获取当前写入的工作表对象（关键：避免KeyError）
        worksheet = writer.sheets[sheet_name]
        
        # 调用自定义格式化函数
        Excel_formatting(worksheet, workbook)

为什么之前的方法会失败？ #

• 如果先一次性写入所有DataFrame，再循环表名去writer.sheets查找，可能因为xlsxwriter的内部逻辑延迟导致工作表未被及时注册（虽然理论上写入后就存在，但即时获取的写法更可靠）
• 表名如果包含特殊字符（如空格、中文），直接查找时容易出现拼写错误，触发KeyError
• 切换到openpyxl引擎时，格式化逻辑与xlsxwriter不兼容，原有的格式化函数无法直接复用，导致看似“失败”

额外提示 #

• 确保自定义格式化函数中使用xlsxwriter专属API（如workbook.add_format()、worksheet.set_row()），不要混用其他引擎的方法
• 如果需要条件格式、合并单元格等复杂操作，直接在Excel_formatting函数中基于传入的worksheet对象扩展即可

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