Pandas样式栏颜色取决于条件？

Question

如果列的style.bar.color属性是基于某种条件计算的，我如何呈现Pandas df？在

示例：

df.style.bar(subset=['before', 'after'], color='#ff781c', vmin=0.0, vmax=1.0)

我希望其中一列（df['before']）保持不变的颜色，另一列（df['after']）计算如下：

^{pr2}$

Answer 1

一种方法是使用pd.IndexSlice为df.style.bar创建子集：

i_pos = pd.IndexSlice[df.loc[(df['after']>df['before'])].index, 'after']
i_neg = pd.IndexSlice[df.loc[~(df['after']>df['before'])].index, 'after']
df.style.bar(subset=['before'], color='#ff781c', vmin=0.0, vmax=1.0)\
  .bar(subset=i_pos, color='green', vmin=0.0, vmax=1.0)\
  .bar(subset=i_neg, color='red', vmin=0.0, vmax=1.0)

输出：