有条件地给 pandas 列上色

Question

我的数据框如下所示：

Week    Item    Buyer   11  12  13  
0   clothes buyerID1    2   3   4   
1   food    buyerID2    2   1   1   
2   water   buyerID     7   5   1

11、12、13 是星期几。我正在尝试有条件地为最后一列的行着色，以描述每周购买的商品的变化。我在这里使用的改编版本，但我运行陷入 IndexingError: Too many indexers 错误。

下面是我修改后的代码：

def highlight3(x):
#if increase
c1 = 'background-color: green'

#if decrease
c2 = 'background-color: red'

c3 = ''

#last row greater than value in second to last row
m1 = x.iloc[:, -1] > x.iloc[:, -2]

#last row lesser than value in second to last row
m2 = x.iloc[:, -1] < x.iloc[:, -2]

out = np.select([m1, m2], [c1, c2], default=c3)
return pd.DataFrame(out, index=x.index, columns=x.columns)

然后我将它应用到我的 df 使用：df.apply(highlight3, axis=None)

Answer 1

解决方法如下：

data = """Week    Item    Buyer   11  12  13  
0   clothes buyerID1    2   3   4   
1   food    buyerID2    2   1   1   
2   water   buyerID     7   5   1   """
df = pd.read_csv(StringIO(data), sep="\s+")

green = 'background-color: green'
red = 'background-color: red'

def style_last_week(x):
    s = pd.Series([""] * x.size)
    if x[-1] > x[-2]:
        s[x.size -1 ] = green

    if x[-2] > x[-1]:
        s[x.size -1 ] = red

    return s

df.style.apply(style_last_week, axis=1)

结果是：

有条件地给 pandas 列上色

Conditionally color pandas column

python

pandas

pandas-apply

pandas-styles