pandas 中的С条件颜色格式

2022-09-04 00:00:00 python pandas styling

问题描述

条件格式中的任务(我想是使用样式) 巨蟒, pandas

有一个有两列的盘子

应突出显示第二个表 条件:

  1. 如果第一个列数超过第二个列数,则为绿色;
  2. 如果第一个列数等于第二个列数,则为黄色;
  3. 如果第一个列数小于第二个列数,则为红色。[

谢谢您的帮助!

推荐答案的想法是使用Styler.apply创建新的按条件填充样式的DataFrame,用于按条件设置行DataFrame.mask
def highlight(x):
    c1 = 'background-color: green'
    c2 = 'background-color: yellow'
    c3 = 'background-color: red'

    m1 = x.iloc[:, 0] > x.iloc[:, 1]
    m2 = x.iloc[:, 0] == x.iloc[:, 1]

    df1 = pd.DataFrame(c3, index=x.index, columns=x.columns)
    return df1.mask(m1, c1).mask(m2, c2)

df.style.apply(highlight, axis=None)

编辑:

如果只需要设置一列,请使用numpy.select

def highlight(x):
    c1 = 'background-color: green'
    c2 = 'background-color: yellow'
    c3 = 'background-color: red'
    c = ''

    m1 = x.iloc[:, 0] > x.iloc[:, 1]
    m2 = x.iloc[:, 0] == x.iloc[:, 1]

    df1 = pd.DataFrame(c, index=x.index, columns=x.columns)
    df1.iloc[:, 1] = np.select([m1, m2], [c1, c2], default=c3)
    return df1

相关文章