首页 > 解决方案 > 保存前着色数据框

问题描述

我创建了一个 df,其中的值为 *"Yes"、"No""Maybe"。*

在保存到 Excel 之前,我想使这些单元格变得丰富多彩。如果我使用库xlwt,我有这样的东西:

sheet2.write(row, col, str('Allowed'), style)

但是,我不想再次打开 Excel,它已经用 pandas 创建,以便通过并用我的颜色填充它。我想已经用 excel 格式的所需颜色保存我的数据框。

我的数据框是这样的:

在此处输入图像描述

我想要的输出是:

在此处输入图像描述

关于如何做到这一点的任何建议?

标签: excelpandasdataframexlwt

解决方案


像这样的东西?

def color_df(val): 
    if val == 'Yes':
        color = 'red'
    elif val == 'Maybe':
        color = 'Orange'
    else:
        color = 'Blue'
    return 'color: %s' % color

df = df.style.applymap(color_df).to_excel(path_var + 'styled.xlsx', engine='xlsxwriter')

推荐阅读