Matplotlib 相关热图中缺少标签

Question

我正在使用 UCI 机器学习存储库中的 abalone dataset。我想使用 matplotlib 和 imshow 显示相关热图。

我第一次尝试，效果很好。绘制并标记了所有数值变量，请参见此处：

fig = plt.figure(figsize=(15,8))
ax1 = fig.add_subplot(111)
plt.imshow(df.corr(), cmap='hot', interpolation='nearest')
plt.colorbar()
labels = df.columns.tolist()
ax1.set_xticklabels(labels,rotation=90, fontsize=10)
ax1.set_yticklabels(labels,fontsize=10)
plt.show()

successful heatmap

后来，我在分类变量上使用了 get_dummies()，如下所示：

df = pd.get_dummies(df, columns = ['sex'])

resulting correlation matrix

所以，如果我重用之前的代码来生成漂亮的热图，应该没问题，对吧？错了！

What dumpster fire is this?

所以我的问题是，我的标签去了哪里，我该如何找回它们？！

谢谢！

Answer 1

要取回您的标签，您可以强制 matplotlib 使用足够的 xticks 以便显示所有标签。这可以通过添加来完成

ax1.set_xticks(np.arange(len(labels)))
ax1.set_yticks(np.arange(len(labels)))

在你的声明之前ax1.set_xticklabels(labels,rotation=90, fontsize=10) 和ax1.set_yticklabels(labels,fontsize=10)。

这将导致以下情节：