python - Pandas Styler 子集列按值

Question

我正在使用以下内容为数据框中的单元格着色：

import seaborn as sns

cm1 = sns.diverging_palette(h_pos=130, h_neg=10, s=99, l=55, n=99, as_cmap=True)

df_s = (df.style
    .background_gradient(cmap=cm1, subset=['col1']))

这成功地将背景渐变应用于col1

但是，我想要以下内容：

df_s = (df.style
    .background_gradient(cmap=cm1, subset=['col1'] < x))

哪个不起作用

这个想法是仅将渐变应用于col1小于的值，并显示未着色x的完整数据框。col1 >= x

似乎应该有一种简单的方法来做到这一点，但我似乎无法将参数转换为正确的格式subset。

在此先感谢您的帮助！

Answer 1

您需要使用pd.IndexSlice：

import seaborn as sns

cm1 = sns.diverging_palette(h_pos=130, h_neg=10, s=99, l=55, n=99, as_cmap=True)
np.random.seed(123)
df = pd.DataFrame(np.random.randint(0,100,(5,5)), columns=[*'ABCDE'])
df.style.background_gradient(cmap=cm1, subset=pd.IndexSlice[df['C']<50, 'C'])

输出：