Python 使用百分位数对数据帧单元格的背景进行着色

您可以在使用quantile和merge_asof的自定义函数中使用styler.apply:

def get_colors(s):
    s = s.astype(float).sort_values()
    return (pd.merge_asof(s.reset_index(), s.quantile(q=percentiles).reset_index())
              .set_index('Date')['index']
              .map(lambda x: f'background-color: {percentile_color.get(x, "")}')
           )

df.style.apply(get_colors)

输出:

请注意，您可以 Select 单元格的 colored颜色 是前一个分位数还是下一个分位数为direction:

def get_colors(s):
    s = s.astype(float).sort_values()
    return (pd.merge_asof(s.reset_index(), s.quantile(q=percentiles).reset_index(),
                         direction='forward')
              .set_index('Date')['index']
              .map(lambda x: f'background-color: {percentile_color.get(x, "")}')
           )

df.style.apply(get_colors)

输出:

而且，对于重复的分位数，您可以使用以下命令强制获取最小的分位数:

def get_colors(s):
    s = s.astype(float).sort_values()
    return (pd.merge_asof(s.reset_index(),
                          s.quantile(q=percentiles).drop_duplicates().reset_index())
              .set_index('Date')['index']
              .map(lambda x: f'background-color: {percentile_color.get(x, "")}')
           )

df.style.apply(get_colors)

输出: