pandas style background gradient both rows and columns

后端 未结 3 1539
一生所求
一生所求 2020-11-29 07:04

The pandas style option to add a background gradient is great for quickly inspecting my output table. However, it is applied either row-wise or columns-wise. Would it be pos

3条回答
  •  栀梦
    栀梦 (楼主)
    2020-11-29 07:30

    Currently you can't set the background_gradient for both the rows/columns simultaneously as pointed by Nickil Maveli. The trick is to customize the 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)
    

提交回复
热议问题