pandas 导出 Excel 文件的时候自动列宽，自动加上边框

somenzz

发布于 2022-10-25 20:43:17

1.9K0

发布于 2022-10-25 20:43:17

尝试过 xlrd、xlwt、openpyxl、xlwings、pandas 来处理 Excel，如果说除了读写 Excel，还要做数据分析，还是 pandas 最好用，大多数情况下，你根本不需要把数据插入数据库，再用 SQL 去做数据分析。

至于 pandas 怎么用，官方网站有个 10 分钟上手 pandas 的教程[1]，没有体验过的可以去体验下。也可以参考 API 说明[2]。

今天主要分享一段代码，可以让 pandas 导出 Excel 文件的时候自动列宽，自动加上边框，省去了手工调整的麻烦。

def to_excel_autowidth_and_border(writer, df, sheetname, startrow,startcol):
    df.to_excel(
        writer, sheet_name=sheetname, index=False, startrow=startrow, startcol=startcol
    )  # send df to writer
    workbook = writer.book
    worksheet = writer.sheets[sheetname]  # pull worksheet object
    formater = workbook.add_format({"border": 1})
    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
                )
            )
            * 3
            + 1
        )  # adding a little extra space
        # print(max_len)
        worksheet.set_column(
            idx + startcol, idx + startcol, max_len
        )  # set column width
    first_row = startrow
    first_col = startcol
    last_row = startrow + len(df.index)
    last_col = startcol + len(df.columns)
    worksheet.conditional_format(
        first_row,
        first_col,
        last_row,
        last_col - 1,
        options={"type": "formula", "criteria": "True", "format": formater},
    )

def main():
    writer = pd.ExcelWriter("./result.xlsx")
    df = pd.read_excel(
        sheet_name=sheet_name, io="/Users/aaron/Desktop/xxxx.xlsx"
    )
    to_excel_autowidth_and_border(writer, df, sheetname="缺陷分析结果", startrow=1, startcol=1)
    writer.save()