如何使用带有斧头上的颜色条的calmap绘制热图？

如何解决如何使用带有斧头上的颜色条的calmap绘制热图？

我有一个熊猫 df 如下：

ipdb> df_calmap
                     count
start_datetime            
2021-03-10 17:40:24      1
2021-03-11 23:41:34      1
2021-03-12 00:41:42      1
2021-03-12 13:11:25      1
2021-03-15 00:02:49      1
2021-03-15 11:40:50      1
2021-03-17 12:42:14      1
2021-03-18 01:50:22      1
2021-03-21 19:10:55      1
2021-03-22 15:00:06      1
2021-03-25 00:22:22      1
2021-03-26 16:33:11      1
2021-03-29 19:16:59      1
2021-03-30 17:58:53      1
2021-03-31 17:46:08      1

我能够使用calmap绘制热图，但是我希望颜色条与图像的大小相同，如对我的问题的类似答案中所示的here。但是，该代码对我不起作用。我似乎无法实现我想要的。

我已经搜索了 matplotlib 文档，但我也无法让它工作。这是我当前的代码。非常感谢。

def plot_trades_heatmap_chart(self):                                                                                                                                                          
    df_calmap = ...                                                                                                                                                                              
                                                                                                                                             
    fig,ax = calmap.calendarplot(df_calmap['count'],monthticks=1,daylabels='MTWTFSS',dayticks=[0,1,2,3,4,5,6],cmap='RdYlGn',fillcolor='whitesmoke',linewidth=0.3,fig_kws=dict(figsize=(8,4)))
                                                                                                                                                                                              
    # Vertical                                                                                                                                                                                
    fig.colorbar(ax[0].get_children()[1],ax=ax.ravel().tolist())                                                                                                                             
                                                                                                                                                                                              
    plt.xlabel("Trades grouped by day",fontsize=12)
    fig.suptitle('Number of trades per day heatmap',fontsize=16)
    return(fig)

预期的输出应该是这样的：

非常感谢！

解决方法

这里有一个 great answer，但它是年份图。我试着看看我是否可以用你想要的日历图做同样的事情，但没有成功。所以我对年份图做了一些改动，并以日历图样式编写了代码。

df_calmap.set_index('start_datetime',inplace=True)

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable

yyyy = 2021
fig = plt.figure(figsize=(20,4))
ax = fig.add_subplot(111)
                                                                                             
cax = calmap.yearplot(df_calmap['count'],year=yyyy)                                                                                                                                                                                           
plt.xlabel("Trades grouped by day",fontsize=12)
plt.ylabel(yyyy,fontsize=58,color='#f5f5f5',weight='bold')
fig.suptitle('Number of trades per day heatmap',fontsize=16)

divider = make_axes_locatable(cax)
lcax = divider.append_axes("right",size="2%",pad=0.5)
fig.colorbar(cax.get_children()[1],cax=lcax)