pandas - 将数据框导出到excel时为单元格着色

Question

我有如下数据框“df”：

x = [1,3,5,7] 
y1 = [3,2,2,2] 
y2 = [2,5,2,2] 
y3 = [7,2,2,1]
df = pd.DataFrame({'x': x, 'y1': y1, 'y2': y2, 'y3': y3})

writer = pd.ExcelWriter('output.xlsx')
df.to_excel(writer,'Sheet1')
writer.save()

我希望 excel 输出文件在 x 列与其他列的相互值中显示相同的颜色。：

Answer 1

如果通过以下方式指定颜色，则可以使用样式dictionary：

def color(a):
    d = {1:'yellow', 3:'green', 5:'blue', 7:'red'}

    d1 = {k: 'background-color:' + v for k, v in d.items()}
    df1 = pd.DataFrame(index=a.index, columns=a.columns)
    df1 = a.applymap(d1.get).fillna('')
    return df1

df.style.apply(color, axis=None).to_excel('styled.xlsx', engine='openpyxl', index=False)