python - Pandas 风格背景渐变行和列

Question

pandas style option添加背景渐变非常适合快速检查我的输出表。但是，它是按行或按列应用的。是否可以一次将其应用于整个数据框？

编辑:一个最小的工作示例:

df = pd.DataFrame([[3,2,10,4],[20,1,3,2],[5,4,6,1]])
df.style.background_gradient()

Answer 1

目前您不能同时为行/列设置background_gradient，如Nickil Maveli 所指出的那样.诀窍是自定义 pandas function background_gradient :

import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import colors

def background_gradient(s, m, M, cmap='PuBu', low=0, high=0):
    rng = M - m
    norm = colors.Normalize(m - (rng * low),
                            M + (rng * high))
    normed = norm(s.values)
    c = [colors.rgb2hex(x) for x in plt.cm.get_cmap(cmap)(normed)]
    return ['background-color: %s' % color for color in c]

df = pd.DataFrame([[3,2,10,4],[20,1,3,2],[5,4,6,1]])
df.style.apply(background_gradient,
               cmap='PuBu',
               m=df.min().min(),
               M=df.max().max(),
               low=0,
               high=0.2)