Pandas造型师如何让Pandas在for循环中显示这个渐变背景?

2024-04-25 02:21:22 发布

您现在位置:Python中文网/ 问答频道 /正文

我有下面的代码,它返回一个对象,如果我存储在一个变量中,我可以通过调用这个变量来显示热图

def heatmaps(var):
    values = user_freq[var].unique()    
    g= []
    for i in values:
        ind = user_freq[user_freq[var] == i]
        g = ind.reset_index().pivot_table(values='timestamp', index=['hour'], columns=['day'], aggfunc='sum', fill_value=0)
    return g.style.background_gradient(cmap='summer')

但我要找的是这样的东西,在那里热图将显示每一次通过。这样我可以给函数一个变量,它会为这个变量的每个值创建一系列热图

    def heatmaps(var):
        values = user_freq[var].unique()    
        g= []
        for i in values:
            ind = user_freq[user_freq[var] == i]
            g = ind.reset_index().pivot_table(values='timestamp', index=['hour'], columns=['day'], aggfunc='sum', fill_value=0)
            g.style.background_gradient(cmap='summer') # DISPLAY THIS IN EACH PASS

        return 

但当然,这不会带来任何回报…你知道我该怎么做吗


Tags: inforindexvardeftablepivot热图