如何在Matplotlib中将任意尺寸图像放入子图且不改变布局对齐？

问题根源在于imshow默认会修改子图坐标轴的宽高比以匹配图像，从而破坏Gridspec预设的布局。以下是两种可靠的解决方式：

方法一：使用OffsetImage（推荐，布局稳定） #

通过将图像作为注释元素添加到子图中，完全避免修改坐标轴的宽高比，同时保持图像原始比例。

完整代码示例 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.offsetbox import OffsetImage, AnnotationBbox

# 创建Gridspec布局（以2行3列为例，可替换为你的自定义布局）
fig = plt.figure(figsize=(10, 6))
gs = fig.add_gridspec(2, 3)

ax1 = fig.add_subplot(gs[0, 0])
ax2 = fig.add_subplot(gs[0, 1])
ax3 = fig.add_subplot(gs[0, 2])
ax4 = fig.add_subplot(gs[1, 0])
ax5 = fig.add_subplot(gs[1, 1])
ax6 = fig.add_subplot(gs[1, 2])

# 隐藏坐标轴刻度（可选，让布局更整洁）
for ax in [ax1, ax2, ax3, ax4, ax5, ax6]:
    ax.set_xticks([])
    ax.set_yticks([])

# 定义按原始比例显示图像的函数
def plot_image(ax, img):
    # 创建保持原始比例的图像容器
    img_box = OffsetImage(img, cmap='viridis', aspect='equal')
    # 将图像居中放置在子图内
    ab = AnnotationBbox(img_box, (0.5, 0.5), xycoords='axes fraction',
                        box_alignment=(0.5, 0.5), frameon=False)
    ax.add_artist(ab)

# 生成随机图像并显示
img1 = np.random.rand(10, 20)
img2 = np.random.rand(15, 25)
img3 = np.random.rand(20, 30)
img4 = np.random.rand(25, 35)
img5 = np.random.rand(30, 40)
img6 = np.random.rand(35, 45)

plot_image(ax1, img1)
plot_image(ax2, img2)
plot_image(ax3, img3)
plot_image(ax4, img4)
plot_image(ax5, img5)
plot_image(ax6, img6)

plt.tight_layout()
plt.show()

如果需要让图像填满子图（超出部分裁剪），可修改plot_image函数，计算合适的缩放比例：

def plot_image_fill(ax, img):
    ax_w, ax_h = ax.get_window_extent().width, ax.get_window_extent().height
    img_w, img_h = img.shape[1], img.shape[0]
    
    # 计算填满子图的缩放比例
    zoom = max(ax_w / img_w, ax_h / img_h)
    img_box = OffsetImage(img, cmap='viridis', zoom=zoom, aspect='equal')
    ab = AnnotationBbox(img_box, (0.5, 0.5), xycoords='axes fraction',
                        box_alignment=(0.5, 0.5), frameon=False)
    ax.add_artist(ab)

方法二：手动调整坐标轴范围 #

通过强制子图保持Gridspec的宽高比，再调整图像显示范围使其居中，保留原始比例：

def plot_image_adjust(ax, img):
    # 获取图像和子图的宽高比
    img_aspect = img.shape[1] / img.shape[0]
    ax_aspect = ax.get_window_extent().width / ax.get_window_extent().height
    
    ax.imshow(img, cmap='viridis', aspect='equal')
    
    # 根据宽高比差异调整坐标轴范围，让图像居中
    if img_aspect > ax_aspect:
        # 图像更宽，上下留白
        y_center = (ax.get_ylim()[0] + ax.get_ylim()[1]) / 2
        new_y_range = (img.shape[0]/img.shape[1]) * ax_aspect * (ax.get_ylim()[1]-ax.get_ylim()[0])
        ax.set_ylim(y_center - new_y_range/2, y_center + new_y_range/2)
    else:
        # 图像更高，左右留白
        x_center = (ax.get_xlim()[0] + ax.get_xlim()[1]) / 2
        new_x_range = (img.shape[1]/img.shape[0]) / ax_aspect * (ax.get_xlim()[1]-ax.get_xlim()[0])
        ax.set_xlim(x_center - new_x_range/2, x_center + new_x_range/2)
    
    # 强制子图保持Gridspec布局的宽高比
    ax.set_aspect('auto')

核心逻辑 #

• 避免imshow修改子图的宽高比：方法一通过注释元素完全脱离坐标轴控制；方法二强制子图使用auto宽高比锁定Gridspec布局。
• 图像自身比例保留：通过aspect='equal'确保图像不被拉伸变形。

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