таблица цветов matplotlib только для одного из участков

Question

Я строю некоторые изображения с помощью matplotlib следующим образом:

fig, axes = plt.subplots(1, len(slices))
    for i, slice in enumerate(slices):
        if i != 3:
            axes[i].imshow(slice.T, cmap="gray", origin="lower")
        else:
            axes[i].imshow(slice.T, cmap="hot", origin="lower")

Как вы можете видеть, одна из осей подплота является тепловой картой. Возможно ли иметь цветовую карту рядом с этим подзаговором?

РЕДАКТИРОВАТЬ:

Хорошо, я могу сделать что-то вроде:

fig.colorbar(im, ax=axes[i])

Это показывает цветную полосу, но она непропорционально велика! Можно ли сделать его такой же высоты, как и остальная часть сюжета.

Answer 1

Хорошо, для кого-то еще вот что сработало:

im = axes[i].imshow(slice.T, cmap="hot", origin="lower")
fig.colorbar(im, ax=axes[i], fraction=0.046, pad=0.04)