python - 有没有办法使用 pandas.ExcelWriter 自动调整 Excel 列宽？

Question

我被要求生成一些 Excel 报告。我目前大量使用 pandas 来处理我的数据，所以我很自然地想使用 pandas.ExcelWriter 方法来生成这些报告。然而，固定的列宽是个问题。

到目前为止，我的代码很简单。假设我有一个名为 df 的数据框:

writer = pd.ExcelWriter(excel_file_path, engine='openpyxl')
df.to_excel(writer, sheet_name="Summary")

我正在查看 pandas 文档，但我并没有看到任何设置列宽的选项。是否有技巧可以使列自动调整到数据？或者我可以对 xlsx 文件做些什么来调整列宽？

(我正在使用 OpenPyXL 库，并生成 .xlsx 文件 - 如果这有什么不同的话。)

Answer 1

灵感来自 user6178746's answer ，我有以下内容:

# Given a dict of dataframes, for example:
# dfs = {'gadgets': df_gadgets, 'widgets': df_widgets}

writer = pd.ExcelWriter(filename, engine='xlsxwriter')
for sheetname, df in dfs.items():  # loop through `dict` of dataframes
    df.to_excel(writer, sheet_name=sheetname)  # send df to writer
    worksheet = writer.sheets[sheetname]  # pull worksheet object
    for idx, col in enumerate(df):  # loop through all columns
        series = df[col]
        max_len = max((
            series.astype(str).map(len).max(),  # len of largest item
            len(str(series.name))  # len of column name/header
            )) + 1  # adding a little extra space
        worksheet.set_column(idx, idx, max_len)  # set column width
writer.save()