python - 在 seaborn 热图上显示日期

Question

我正在尝试使用 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 轴上没有打印任何内容。

Answer 1

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()