如何使用Styler函数生成多个PNG文件？解决OSError报错

错误原因 #

你遇到的OSError源于两个核心问题：

1. 循环结构错误：内层导出循环缩进在第一个循环内部，导致每次向test列表添加一个Styler对象后就立刻遍历test导出，且变量i未定义，Python解析文件名时误将Styler对象的字符串表示代入，生成了包含<pandas.io.formats.style.Styler object...>的非法文件名。
2. 文件名变量未正确设置：没有生成合法的递增序号来区分不同的PNG文件。

修正后的代码 #

import pandas as pd
import dataframe_image as dfi

# 初始化数据
df = pd.DataFrame(data=[[-100,500,400,0,222,222], [9000,124,0,-147,54,-56],[77,0,110,211,0,222], [111,11,-600,33,0,22],[213,-124,0,-147,54,-56]])
df2 = pd.DataFrame(data=[[100,500,200,0,555,222], [5000,124,0,-147,54,50],[77,0,500,211,0,222], [-900,11,-600,33,0,22],[500,-124,0,-147,54,-56]])

# 设置多级列名
df.columns = pd.MultiIndex.from_product([['x','y','z'], list('ab')])
df2.columns = pd.MultiIndex.from_product([['x','y','z'], list('ab')])

df_list = [df, df2]

# 样式函数
def colors(i):
    if i > 0:
        return 'background: red'
    elif i < 0:
        return 'background: green'
    elif i == 0:
        return 'background: yellow'
    else:
        return ''  # 补充返回空字符串，避免返回None

test = []
# 第一步：批量生成带样式的Styler对象
for df in df_list:
    styled_df = df.style.applymap(colors, subset=pd.IndexSlice[:, pd.IndexSlice[:,'a']])
    test.append(styled_df)

# 第二步：按序号导出每个Styler对象为PNG
for idx, styled_df in enumerate(test):
    # 用索引生成唯一文件名，比如test1.png、test2.png
    dfi.export(styled_df, f'test{idx+1}.png')

关键修正说明 #

• 调整循环结构：将导出循环移到生成Styler对象的循环外部，确保所有样式对象准备好后再批量导出，避免重复操作。
• 合法文件名生成：使用enumerate获取每个对象的索引，以此作为文件名的序号，保证文件名唯一且符合系统规范。
• 修复样式函数：补充else分支的返回值，避免因返回None导致的样式异常。

内容的提问来源于stack exchange，提问作者user18708380