求助：python中用plt绘制图形时，如何限定图形尺寸（要使横纵坐标不等比例）？

Question

需要将矩阵可视化，生成一个正方形的热力图，但原矩阵不是方阵，导致输出的图形始终是个矩形。

尝试了plt.figure(figsize=(6, 6))；但只能让输出的图片是正方形（绘图区还是一个矩形，结果见后图）。

恳请大神赐教，不胜感激！

---

import numpy as np
import matplotlib.pyplot as plt

#data为8x4的矩阵
data=np.array([[1,3,5,2],[2,6,4,9],[4,7,0,3],[8,5,1,7],[1,3,6,2],[2,8,4,0],[3,7,9,1],[8,5,1,7]])

plt.figure(figsize=(6, 6))
cmap=plt.cm.inferno#设置热力图的颜色样式
img=plt.imshow(data,cmap=cmap)
plt.colorbar()

Answer 1

plt.axis('tight')