为什么使用“to_excel”保存时熊猫数据框样式丢失？

Question

根据此示例，该to_excel方法应保存带有背景颜色的 Excel 文件。但是，我保存的 Excel 文件中没有任何颜色。我尝试同时使用openpyxl和xlsxwriter引擎进行编写。在这两种情况下，Excel 文件都保存了，但单元格颜色/样式丢失了。

我可以读回文件并使用 重新格式化openpyxl，但如果这种to_excel方法应该有效，为什么不呢？

这是示例代码。

 import pandas as pd # version 0.24.2
 dict = {'A': [1, 1, 1, 1, 1], 'B':[2, 1, 2, 1, 2], 'C':[1, 2, 1, 2, 1]}
 df = pd.DataFrame(dict)
 df_styled = df.style.apply(lambda x: ["background: #ffa31a" if x.iloc[0] < v else " " for v in x], axis=1)

 df_styled 
 ''' in my jupyter notebook, this displayed my dataframe with background color when condition is met, (all the 2s highlighted)'''

 '''Save the styled data frame to excel using to_excel'''
 df_styled.to_excel('example_file_openpyxl.xlsx', engine='openpyxl')
 df_styled.to_excel('example_file_xlsxwriter.xlsx', engine='xlsxwriter')

Answer 1

我自己偶然发现了这一点，据我所知，尚不支持像这样导出到 Excel。我已经调整了您的代码以匹配文档中的 excel 输出。

这是文档输出到excel的方法。

df.style.\
    applymap(color_negative_red).\
    apply(highlight_max).\
    to_excel('styled.xlsx', engine='openpyxl')

这是您调整后的代码：

import pandas as pd
dict = {'A': [1, 1, 1, 1, 1], 'B':[2,1,2,1,2], 'C':[1,2,1,2,1]}
df = pd.DataFrame(dict)

def highlight(df, color = "yellow"):

    attr = 'background-color: {}'.format(color)
    df_bool = pd.DataFrame(df.apply(lambda x: [True if x.iloc[0] < v else False for v in x],axis=1).apply(pd.Series),
                      index=df.index)
    df_bool.columns =df.columns
    return pd.DataFrame(np.where(df_bool, attr, ""),
                       index= df.index, columns=df.columns)
df.style. \
    apply(highlight, axis=None).\
    to_excel("styled.xlsx", engine="openpyxl")

在突出显示函数内，我根据上面列表理解中应用的条件创建一个布尔数据框。然后，我根据该数据框的结果分配样式。