How can I keep cells square in heatmap?

Question

When the cbar is on, the cells in seaborn's heatmap are rectangular. I could change the figsize of my axis, but I am wondering if there is an easier way to keep the cells square.

Answer 1

seaborn 热图有选项square ：

heatmap(data, square=True)

Answer 2

You can use ax.set_aspect("equal") to set an equal aspect ratio for the axes ax .

Example:

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()