Используйте applymap
с get
по dict для DataFrame
для цветов и передайте Styler.apply
:
df1 = pd.DataFrame({
'B':[4,5,4],
'C':[7,8,9],
'D':[1,3,5],
})
df2 = pd.DataFrame({
'B':[1,np.nan,4],
'C':[np.nan,2,np.nan],
'D':[1,3,np.nan],
})
def apply_color(x):
colors = {1: 'green',2: 'blue', 3: 'yellow', 4: 'orange', 5: 'grey'}
return df2.applymap(lambda val: 'background-color: {}'.format(colors.get(val,'')))
df1.style.apply(apply_color, axis=None)