定制Pandas导出数据表的式样

Pandas 是 Python 中一个常用的数据处理库，不过其输出的数据样式较为简单。本文介绍了如何使用 StyleFrame 包来定制和优化 pandas 导出的 Excel 数据样式。

StyleFrame 介绍

StyleFrame 是一个 Python 包，专门用于在 Excel 中创建、编辑和格式化大型数据集。它提供了多种功能使得数据在 Excel 表格中看起来更美观和易读。

安装

通过运行 pip install styleframe 可以轻松安装 StyleFrame。

生成数据

首先导入 StyleFrame 和相关类，然后使用 pandas 创建 DataFrame。接着，使用 StyleFrame 包装这个 DataFrame。

    
from styleframe import StyleFrame, Styler, utils
import pandas as pd
df1 = pd.DataFrame({
  'col_a': range(20),
  'col_b': [i for i in range(20000000,20000000+20)],
  'col_c': range(20)
})
sf = StyleFrame(df1)
    
  

设置式样

StyleFrame 允许用户自定义每一列的样式、表头的样式以及列宽。

    
# 设置每一列的配置
sf.apply_style_by_indexes(
  indexes_to_style=sf[sf['col_a'] > 10],
  cols_to_style=['col_a'],
  styler_obj=Styler(
    bg_color=utils.colors.blue,
    bold=True,
    font_size=10
  )
)
# ... 其他列的样式设置 ...
# 设置列宽
sf.set_column_width_dict({
  'col_a': 10,
  'col_b': 30,
  'col_c': 10,
})
# 设置表头式样
header_style = Styler(bg_color="yellow", bold=True, font_size=10)
sf.apply_headers_style(header_style)
    
  

官方文档提供了完整的 API 及其配置选项，包括丰富的颜色配置。

导出数据

最后，数据可以被导出到一个 Excel 文件中。

    
ew = StyleFrame.ExcelWriter('my_excel.xlsx')
sf.to_excel(ew)
ew.close()
    
  

参考文档链接：StyleFrame 官方文档