如何在matplotlib.pyplot.imshow中指定实际显示宽高比？

问题本质 #

imshow的aspect参数不是直接设置显示的物理宽高比，而是数据维度和显示比例的关联系数，计算公式为：
显示高度/显示宽度 = (y轴数据范围 / x轴数据范围) × aspect

拿你的示例来说，z = np.random.rand(10,1000)对应10行（y轴范围0_{9）、1000列（x轴范围0}999），设置aspect=1时，显示高度/宽度 = (10/1000)×1 = 0.01，也就是宽度是高度的100倍，自然呈现细长图像。

优雅解决方法 #

1. 一键生成正方形显示

无需手动计算行列数，利用坐标轴的get_data_ratio()自动获取数据的y/x范围比，反向设置aspect即可：

import numpy as np
import matplotlib.pyplot as plt

z = np.random.rand(10,1000)
ax = plt.gca()
ax.imshow(z)
# 强制显示为正方形
ax.set_aspect(1 / ax.get_data_ratio())
plt.show()

哪怕之后修改xlim（比如ax.set_xlim(0, 500)），只需重新执行ax.set_aspect(1 / ax.get_data_ratio())，就能保持正方形显示，完全不用手动调整参数。

2. 指定任意固定宽高比

如果想设置其他比例（比如宽度是高度的2倍），可以按以下方式计算：

target_ratio = 2  # 显示宽度 ÷ 显示高度 = 2
ax.set_aspect( (1 / target_ratio) / ax.get_data_ratio() )

原理是通过target_ratio反推需要的aspect值，同样能自动适配数据范围的变化。

注意点 #

• ax.get_data_ratio()会实时返回当前坐标轴的y轴数据范围 ÷ x轴数据范围，不管你怎么调整坐标轴范围，它都能准确计算。
• 这种方法比手动提取行列数灵活得多，尤其适合动态调整数据或坐标轴的场景。

内容的提问来源于stack exchange，提问作者Tom