应用applemap方法对细胞进行染色

Question

我想用panda根据一些单元格的内容给它们上色(在数据帧中(，我做了一些尝试，但没有要求的结果

这是我最后一次失败的尝试：

import pandas as pd
import dataframe_image as dfi
df = pd.read_excel('splice traitment.xlsx', sheet_name='Sheet4', usecols="B,C,D,E,F,G,H,I,J,K")
def color_cells(val):
color = 'red' if val == 7 else ''
return 'background-color: {}'.format(color)
df.style.applymap(color_cells)
dfi.export(df,"table.png")

非常感谢

Answer 1

您可以将样式化的DataFrame传递给export方法：

dfi.export(df.style.applymap(color_cells),"table.png")

或根据变量styled:

styled = df.style.applymap(color_cells)
dfi.export(styled,"table.png")