9

根据此示例,该to_excel方法应使用背景颜色保存 Excel 文件。但是,我保存的 Excel 文件中没有任何颜色。我尝试同时使用openpyxlxlsxwriter引擎来编写。在这两种情况下,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')

4

1 回答 1

4

我自己偶然发现了这个,据我所知,还不支持像这样导出到 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")

在 highlight 函数中,我根据上面列表理解中应用的条件创建了一个布尔数据框。然后,我根据这个数据框的结果分配样式。

于 2019-03-22T21:59:56.367 回答