1

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

在此处输入图像描述

4

2 回答 2

11

Seabornheatmap是一个分类图。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()

在此处输入图像描述

于 2017-07-27T11:15:13.597 回答
0

我在尝试做类似的事情时发现了这个问题，你可以拼凑出一个解决方案，但它不是很漂亮。

例如，我获取当前标签，遍历它们以找到一月份的标签并将其设置为仅年份，将其余标签设置为空白。

这给了我正确位置的年份标签。

xticklabels = ax.get_xticklabels()

for label in xticklabels:
    text = label.get_text()
    if text[5:7] == '01':
        label.set_text(text[0:4])
    else:
        label.set_text('')

ax.set_xticklabels(xticklabels)

希望从中你可以弄清楚你想做什么。

于 2019-03-12T19:48:53.693 回答