Python 3中使用XlWriter多条件设置列宽的异常问题咨询

你遇到的问题核心是Excel列索引和DataFrame列索引之间的偏移没有统一处理，再加上xlsxwriter的set_column方法的索引规则，导致了列宽调整错位。咱们一步步拆解逻辑：

1. 核心逻辑解析 #

首先明确两个关键点：

• enumerate(df)返回的idx是DataFrame自身的列索引（从0开始），比如df的第一列对应idx=0，第二列idx=1，以此类推。
• 如果你用df.to_excel(worksheet, index=True)（这是默认行为），Excel的A列会被用来存放DataFrame的索引，而DataFrame的第一列会对应Excel的B列（即Excel列索引为1）。这就产生了1位的偏移量——DataFrame的列索引idx对应的Excel列索引是idx+1。

你之前的「按索引设置」代码已经考虑了这个偏移（用了idx+1），但「按列名设置」的代码没有统一这个规则，所以出现了错位：

• 当你用worksheet.set_column(idx, idx, 50)：这里的idx是DataFrame的列索引，直接对应Excel的列索引，相当于没加偏移，所以会调整到前一列（比如df的SUMMARY列是idx=1，对应Excel的B列，而非你预期的C列）。
• 当你用worksheet.set_column(idx, idx+1, 50)：这会选中从Excel列idx到idx+1的两列，自然会覆盖B和C列。
• 当你用idx+1, idx+1：相当于加上了偏移，此时df的SINGLE_SENTENCE列（假设idx=2）对应Excel的2+1=3，也就是D列，这才和你的预期匹配。

另外，你提到的「设置被覆盖」是正常现象：xlsxwriter的set_column是最后一次设置生效，如果你先按索引设置了所有列的宽度，再按列名调整特定列，后面的设置会覆盖前面的，这不是bug，是执行顺序导致的预期结果。

2. 解决方案：统一偏移量，用列名映射调整 #

要解决这个问题，我们需要统一处理偏移量，并明确列名到Excel列索引的映射。推荐的写法如下：

# 第一步：先确定偏移量（如果写入Excel时包含DataFrame索引，offset=1；否则offset=0）
offset = 1  # 对应index=True的情况，这是默认值

# 第二步：创建列名到DataFrame列索引的映射字典
col_to_df_idx = {col: idx for idx, col in enumerate(df.columns)}

# 第三步：按列名设置列宽，统一加上偏移量
# 定义需要调整的列名和对应宽度
column_widths = {
    'SINGLE_SENTENCE': 50,
    'SUMMARY': 50,
    'TOTAL': 15
}

for col_name, width in column_widths.items():
    # 获取该列在DataFrame中的索引
    df_idx = col_to_df_idx[col_name]
    # 转换为Excel的列索引（加上偏移）
    excel_col = df_idx + offset
    # 设置单列宽度
    worksheet.set_column(excel_col, excel_col, width)

3. 额外注意事项 #

• 如果你写入Excel时用了index=False，记得把offset改成0，这样DataFrame的列索引就直接对应Excel的列索引了。
• 建议把所有列宽调整的代码放在一起，按「先设置默认宽度，再覆盖特定列」的顺序执行，避免混乱。

内容的提问来源于stack exchange，提问作者Peter.k