Pandas style.bar 颜色基于条件？

Question

如何呈现 Pandas df，其中其中一列的 style.bar.color 属性是根据某些条件计算的？

示例：

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

我不想让两列都用 #ff781c 突出显示，而是希望其中一列 (df['before']) 保持相同的恒定颜色，而另一列 (df['after'] ) 计算为：

def compute_color(row):
   if row['after'] >= row['before']:
      return 'red'
   else:
      return 'green

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)

输出：

Answer 2

为列中的每个单元格明确着色。

    rows = 10
    indx = list(df.index)[-rows:]  # indices of the last 10 rows
    # Colormap for the last 10 rows in a Column
    last10 = df['Column'][-rows:]  # values to color
    colors = [color_map_color(e, cmap_name='autumn_r', vmin=100, vmax=1000) for e in last10]  # colors
    values = [pd.IndexSlice[indx[i], 'Column'] for i in range(rows)]  # for .bar subset

    html = (df.style
            .bar(subset=values[0], color=colors[0], vmax=1000, vmin=0, align='left', width=100)
            .bar(subset=values[1], color=colors[1], vmax=1000, vmin=0, align='left', width=100)
            .bar(subset=values[2], color=colors[2], vmax=1000, vmin=0, align='left', width=100)

            )
    html

https://imgur.com/b2MHw8b

Pandas style.bar 颜色基于条件？

Pandas style.bar color based on condition?

python

pandas

pandas-styles