如何使用Matplotlib与GridSpec使3×3子图显示尺寸一致

嗨，我来帮你解决这个问题！你遇到的核心问题是：不同分辨率的正方形图像（256×256、200×200、128×128）在默认GridSpec布局里，会因为子图高度一致而显示出不同的视觉大小。要让它们看起来一样大，关键是让子图的高度和图像的分辨率成正比，同时保持图像的等比例显示。

这里有个直接有效的解决方案，利用GridSpec的height_ratios参数控制每行子图的高度比例，配合设置子图的等比例显示：

完整修改后的代码示例 #

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
import numpy as np

# 生成测试用的不同尺寸图像（替换成你的img1-img9即可）
img1 = np.random.rand(256, 256)
img2 = np.random.rand(256, 256)
img3 = np.random.rand(256, 256)
img4 = np.random.rand(200, 200)
img5 = np.random.rand(200, 200)
img6 = np.random.rand(200, 200)
img7 = np.random.rand(128, 128)
img8 = np.random.rand(128, 128)
img9 = np.random.rand(128, 128)

fig = plt.figure(figsize=(30, 40))
fig.patch.set_facecolor('white')

# 关键设置：让每行子图高度比例和对应图像的高度一致
gs1 = gridspec.GridSpec(3, 3, height_ratios=[256, 200, 128])
gs1.update(wspace=0.05, hspace=0.05)

# 创建子图
ax1 = plt.subplot(gs1[0])
ax2 = plt.subplot(gs1[1])
ax3 = plt.subplot(gs1[2])
ax4 = plt.subplot(gs1[3])
ax5 = plt.subplot(gs1[4])
ax6 = plt.subplot(gs1[5])
ax7 = plt.subplot(gs1[6])
ax8 = plt.subplot(gs1[7])
ax9 = plt.subplot(gs1[8])

# 批量处理图像显示，确保比例一致并优化视觉效果
axes = [ax1, ax2, ax3, ax4, ax5, ax6, ax7, ax8, ax9]
images = [img1, img2, img3, img4, img5, img6, img7, img8, img9]

for ax, img in zip(axes, images):
    ax.imshow(img, cmap='gray')
    ax.set_aspect('equal')  # 强制图像保持正方形比例，避免拉伸
    ax.axis('off')  # 可选：隐藏坐标轴，让图像视觉大小对比更清晰

plt.show()

原理说明 #

1. GridSpec的height_ratios：我们把每行子图的高度设置成和对应图像的高度一致（256、200、128），这样第一行子图的高度会比第二行高，第二行比第三行高，刚好抵消图像分辨率的差异，让不同尺寸的图像显示出相同的视觉大小。
2. set_aspect('equal')：确保图像在子图里保持原有的正方形比例，不会被拉伸变形。
3. axis('off')：隐藏坐标轴可以消除额外的视觉干扰，让整体布局更整洁，也能更直观地看到图像大小的一致性。

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