如何避免matplotlib的Axes.imshow破坏绘图布局？

取消注释ax.imshow()后布局被破坏，核心原因是imshow默认启用**aspect='equal'**参数——它会强制坐标轴的宽高比例匹配图像数据的宽高比（你的数据是2x2正方形），而原本画布是A4尺寸（8.5×11，宽高比≈0.77），这种强制匹配会打乱坐标轴铺满画布的原有布局，导致最终图像尺寸异常。

要维持A4画布尺寸和坐标轴铺满画布的布局，同时在指定区域绘制图像，只需在imshow中显式设置**aspect='auto'**，让图像自适应指定区域的比例：

修正后的代码 #

import matplotlib.pyplot as plt
import numpy as np

papersize = (8.5, 11)

fig = plt.figure(figsize=papersize)
ax = fig.add_axes(rect=(0, 0, 1, 1))
ax.add_patch(plt.Rectangle((0, 0), 1, 1, color="red", zorder=-1))

# 启用aspect='auto'，让图像自适应指定区域，不强制纵横比
ax.imshow(np.asarray([[0, 1], [1, 0]]), cmap="gray", extent=(0.25, 0.75, 0.25, 0.75), aspect='auto')

# 绘制多幅图像示例：重复调用imshow，设置不同extent即可
# ax.imshow(np.asarray([[1, 0], [0, 1]]), cmap="gray", extent=(0.1, 0.4, 0.1, 0.4), aspect='auto')
# ax.imshow(np.asarray([[0, 0], [1, 1]]), cmap="gray", extent=(0.6, 0.9, 0.6, 0.9), aspect='auto')

fig.savefig("test.png", dpi=200)

核心说明 #

• aspect='auto'：解除imshow对坐标轴纵横比的强制约束，让图像拉伸填满你通过extent指定的区域，同时保留原画布的A4尺寸和坐标轴铺满整个画布的布局。
• 多图像绘制：多次调用ax.imshow，为每幅图像设置独立的extent参数（格式为(xmin, xmax, ymin, ymax)）即可指定各自的位置和大小，所有图像都会自适应区域，不会破坏整体布局。

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