python之在 seaborn 热图上显示日期

我正在尝试使用 seaborn 库从 pandas 数据框创建热图。这是代码:

test_df = pd.DataFrame(np.random.randn(367, 5),  
                 index = pd.DatetimeIndex(start='01-01-2000', end='01-01-2001', freq='1D')) 
 
ax = sns.heatmap(test_df.T) 
ax.xaxis.set_major_locator(mdates.MonthLocator()) 
ax.xaxis.set_minor_locator(mdates.DayLocator()) 
ax.xaxis.set_major_formatter(mdates.DateFormatter('%b')) 
ax.xaxis.set_minor_formatter(mdates.DateFormatter('%d')) 

但是，我得到的图形在 x 轴上没有打印任何内容。

请您参考如下方法:

Seaborn heatmap 是一个分类图。它从 0 到 number of columns - 1，在本例中从 0 到 366。日期时间定位器和格式化程序期望值为日期(或更准确地说，对应于日期的数字)。对于所讨论的年份，这将是 730120 (= 01-01-2000) 和 730486 (= 01-01-2001) 之间的数字。

因此，为了能够使用 matplotlib.dates 格式化程序和定位器，您需要先将数据帧索引转换为日期时间对象。然后您可以不使用热图，而是使用允许数字轴的图，例如imshow 图。然后，您可以设置该 imshow 图的范围以对应于您要显示的日期范围。

import pandas as pd 
import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.dates as mdates 
 
df = pd.DataFrame(np.random.randn(367, 5),  
                 index = pd.DatetimeIndex(start='01-01-2000', end='01-01-2001', freq='1D')) 
 
dates = df.index.to_pydatetime() 
dnum = mdates.date2num(dates) 
start = dnum[0] - (dnum[1]-dnum[0])/2. 
stop = dnum[-1] + (dnum[1]-dnum[0])/2. 
extent = [start, stop, -0.5, len(df.columns)-0.5] 
 
fig, ax = plt.subplots() 
im = ax.imshow(df.T.values, extent=extent, aspect="auto") 
 
ax.xaxis.set_major_locator(mdates.MonthLocator()) 
ax.xaxis.set_minor_locator(mdates.DayLocator()) 
ax.xaxis.set_major_formatter(mdates.DateFormatter('%b')) 
 
fig.colorbar(im) 
plt.show()