保存前为数据框着色

时间:2018-10-16 09:23:50

标签: excel pandas dataframe xlwt

我创建了一个df,其值为*“ Yes”,“ No” “ Maybe”。*

在保存excel之前,我想使这些单元格丰富多彩。如果我使用库 xlwt ,我将遇到以下情况:

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

但是,我不想再次打开excel,它已经用熊猫创建了,以便通过我的颜色填充它。我想使用在excel中格式化的所需颜色来保存数据框。

我的数据框是这样的:

enter image description here

我想要的输出是:

enter image description here

有关如何执行此操作的任何建议?

1 个答案:

答案 0 :(得分:2)

像这样吗?

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')