我有两个具有相同索引/列的数据帧:
df = pd.DataFrame({'A':[5.5, 3, 0, 3, 1],
'B':[2, 1, 0.2, 4, 5],
'C':[3, 1, 3.5, 6, 0]})
df_bool = pd.DataFrame({'A':[0, 1, 0, 0, 1],
'B':[0, 0, 1, 0, 0],
'C':[1, 1, 1, 0, 0]})
我想使用df_bool作为掩码,在df元素方面应用一个样式函数.
这是预期的结果:
当前失败的功能
def color_boolean(val):
color =''
if df_bool == 1:
color = 'red'
elif df_bool == 0:
color = 'black'
return f'color: {color}'
df.head().style.apply(color_boolean, axis=None)
ValueError: The truth value of a DataFrame is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().