pandas DataFrame 导出到Excel格式美化

stone0823

已于 2024-02-21 20:58:48 修改

阅读量874

点赞数 9

分类专栏： # pandas 文章标签： pandas excel

于 2024-02-20 23:37:56 首次发布

本文链接：https://blog.csdn.net/stone0823/article/details/136200856

版权

pandas 专栏收录该内容

11 篇文章 21 订阅

订阅专栏

本文介绍了如何使用pandas的to_excel()方法配合xlsxwriter库，通过自动调整列宽和创建Excel表格（超级表）来美化DataFrame的Excel导出格式，实现了一个通用化的代码示例。

摘要由CSDN通过智能技术生成

默认情况下，DataFrame 的 to_excel() 方法导出到 Excel 格式是比较丑的，本篇的代码演示了用一种比较简单的方法进行美化。要点：

to_excel() 方法的引擎使用 xlsxwriter
使用 Excel 的 Table （经常被称为超级表）
自动列宽

美化前，输出的格式如下：

美化后格式如下：

代码实现了通用化：


import pandas as pd
import os

def format_excel(writer, sheet_name, df):
    '''
    将DataFrame写入Excel, 格式美化
    writer:     XlsxWriter
    sheet_name: 要写入的工作表名
    df:         DataFrame
    '''
    column_settings = [{'header': column} for column in df.columns]
    (max_row, max_col) = df.shape

    worksheet = writer.sheets[sheet_name]
    worksheet.add_table(0, 0, max_row, max_col - 1, {'columns': column_settings,'banded_rows': False })
    worksheet.autofit()

    writer.close()


path = os.path.dirname(os.path.abspath(__file__))
fname = os.path.join(path, 'source.xlsx')

df = pd.read_excel(fname, sheet_name=0)

writer = pd.ExcelWriter(os.path.join(path, 'output.xlsx'), engine='xlsxwriter')
df.to_excel(writer, 'Sheet1', index=False)
format_excel(writer, "Sheet1", df)