如何在热图中保持单元格方形？

Question

当 cbar 开启时，seaborn 热图中的单元格是矩形的。我可以改变我的轴的 figsize，但我想知道是否有更简单的方法来保持单元格的正方形。

Answer 1

seaborn 热图有选项square：

heatmap(data, square=True)

Answer 2

您可以使用ax.set_aspect("equal")为轴设置相等的纵横比ax。

例子：

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

data = np.random.rand(4,16)
ax = sns.heatmap(data)
ax.set_aspect("equal")
plt.show()