Подграфики создают графики разной высоты ⇐ Python

[Anonymous]

Я хочу создать фигуру, содержащую три разных участка тепловой карты. Все три субблота должны быть выровнены по строкам, поскольку они имеют общее значение. Но ширина каждого подграфика разная, и при их объединении я получаю разную высоту, поэтому строки не выравниваются.
Как это решить?
Вот минимальный пример кода и картинка [1].

Код: Выделить всё

import numpy as np
from matplotlib.gridspec import GridSpec
rows = 4
first_cols = 5
second_cols = 1
third_cols = 3
first = np.random.randint(0,2,( rows , first_cols ))
second = np.random.randint(0,2,( rows , second_cols ))
third = np.random.randint(0,4,( rows , third_cols ))

fig=plt.figure()

gs=GridSpec(1,first_cols+second_cols+third_cols)

ax1=fig.add_subplot(gs[0,0:first_cols])
ax2=fig.add_subplot(gs[0,first_cols:first_cols+second_cols])
ax3=fig.add_subplot(gs[0,first_cols+second_cols:])
im1 = ax1.imshow(first)
im2 = ax2.imshow(second)
im3 = ax3.imshow(third)

plt.show()

Подробнее здесь: https://stackoverflow.com/questions/790 ... ight-plots