Matplotlib 相关热图中缺少标签
Missing labels in Matplotlib correlation heatmap
我正在使用来自 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?
所以我的问题是，我的标签去了哪里，我该如何取回它们？！
谢谢！
要恢复标签，您可以强制 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).
之前
这导致以下情节：
我正在使用来自 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?
所以我的问题是，我的标签去了哪里，我该如何取回它们？！
谢谢！
要恢复标签，您可以强制 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).
这导致以下情节：