Как изменить высоту сетки каждого изображения с помощью mpl_toolkits.axes_grid1.Imagegrid

Я рисую изображение с сетками 4 * 6 с помощью mpl_toolkits.axes_grid1.Imagegrid:

target = np.reshape(target, (12, 41, 81))
output = np.reshape(output, (12, 41, 81))
pred_error = target - output
# target: C x D x H x W
sfmt = ticker.ScalarFormatter(useMathText=True)
sfmt.set_powerlimits((-2, 2))
cmap = 'jet'
fig = plt.figure(1, (11, 5.5))
axes_pad = 0.1
cbar_pad = 0.1
label_size = 6
plt.rcParams["mpl_toolkits.legacy_colorbar"] = False
subplots_position = [(4,6,i) for i in range(1, 25)]
for i, subplot_i in enumerate(subplots_position):
    if i in [i for i in range(6)]+[i for i in range(12, 18)]:
        # share one colorbar
        grid = ImageGrid(fig, subplot_i,          # as in plt.subplot(111)
                          nrows_ncols=(2, 1),
                          axes_pad=axes_pad,
                          share_all=False,
                          cbar_location="right",
                          cbar_mode="single",
                          cbar_size="3%",
                          cbar_pad=cbar_pad,
                          )
        if i <6:
            data = (target[i], output[i])
        else:
            data = (target[i-6], output[i-6])
        channel = np.concatenate(data)
        vmin, vmax = np.amin(channel), np.amax(channel)
        # Add data to image grid
        for j, ax in enumerate(grid):
            im = ax.imshow(data[j], vmin=vmin, vmax=vmax, cmap=cmap)
            ax.set_axis_off()
        # ticks=np.linspace(vmin, vmax, 10)
        #set_ticks, set_ticklabels
        cbar = grid.cbar_axes[0].colorbar(im, format=sfmt)
        # cbar.ax.set_yticks((vmin, vmax))
        cbar.ax.yaxis.set_offset_position('left')
        cbar.ax.tick_params(labelsize=label_size)
        cbar.ax.toggle_label(True)

    else:
        grid = ImageGrid(fig, subplot_i,  # as in plt.subplot(111)
                          nrows_ncols=(1, 1),
                          axes_pad=axes_pad,
                        #  share_all=True,
                        #  aspect=True,
                          cbar_location="right",
                          cbar_mode="single",
                          cbar_size="6%",
                          cbar_pad=cbar_pad,
                          )
        data = [pred_error[i%12]]
        for j, ax in enumerate(grid):
            im = ax.imshow(data[j], cmap=cmap)
            ax.set_axis_off()
            ax.set_axes_locator
            cbar = grid.cbar_axes[j].colorbar(im, format=sfmt)
            grid.cbar_axes[j].tick_params(labelsize=label_size)
            grid.cbar_axes[j].toggle_label(True)

plt.tight_layout()##pad=0.25, w_pad=0.25, h_pad=0.25)
# fig.subplots_adjust(wspace=0.075, hspace=0.075)
plt.show()
plt.savefig('test.pdf', 
            dpi=300, bbox_inches='tight')

plt.show()
plt.close(fig)

target – это (2,6,41,81) массив форм, размер которого при построении изменен до (12,41,81), выход имеет то же измерение, pred_error — это разница между ними. Я хочу показать target в 1-й и 4-й строках, output во 2-й и 5-й строках с одной и той же цветовой полосой, pred_error в 3-й и 6-й строках .

Изображение, которое я получаю сейчас: я прокомментировал поля сетки, < img src="https://i.stack.imgur.com/YOPLD.png" alt="красные поля - это моя аннотация, каждое поле представляет собой сетку, я рисую ее таким образом, чтобы первые две строки могли иметь общую цветовую полосу " />

Изображение, которое я хочу, без огромных пробелов:

Я знаю, что проблема в том, что все сетки на моем изображении имеют одинаковый размер, но я не нашел способа отредактировать высоту этих зеленых сеток изображений. Я ценю вашу помощь!!!