Pandas Styler 子集列按值

Question

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

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的值，并显示完整的 dataframe ，其中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'])

Output：