为什么热图颜色值显示为细长的矩形条而不是正方形？ [复制]

Question

如何改善此热图的外观？为什么颜色值显示为细长的矩形条（中间有所有空白）而不是正方形，或者这取决于数据文件通常会发生什么？这不是热图应该的样子。

Data

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
sns.set(color_codes=True)
np.random.seed(sum(map(ord, "distributions")))

data = pd.read_csv('dvAng_heatmap.dat',sep='\s',engine='python')
data2 = pd.pivot_table(data.round(2),values='k',index='g',columns='h')

mask = np.isnan(data2)

sns.set(style="white")

xtics = 20
ytics = 10
cmap = "jet"
vmin = None
vmax = None

ax = sns.heatmap(data2, xticklabels=xtics, yticklabels=ytics, mask=mask, linewidths=0, cbar=True, robust=False,cmap=cmap, vmin=vmin, vmax=vmax)

ax.invert_yaxis()

plt.title('Heatmap')
plt.xlabel('angle')
plt.ylabel('separation')

plt.savefig('Heatmap.png', transparent=True)

sns.plt.show()

Answer 1

阅读documentation。您所要做的就是添加square=True 关键字。默认情况下，此选项设置为 False。

在后台，seaborn 调用 matplotlib.pyplot.imshow 并使用 square 关键字设置 Axes.set_aspect('equal')。如果使用square=True 的效果过于极端，您可以尝试手动设置纵横比：ax.set_aspect(num) 其中num 是描述生成的矩形的高宽比的数字。