Проблема Matplotlib с Colorbar с использованием subplot_mosaic и make_axes_locatable ⇐ Python

[Anonymous]

Я хочу создать фигуру с использованием subplot_mosaic, которая очень полезна для организации графиков. Однако с моим кодом, когда я график фигуру, я не вижу значения и заголовка цвета. Вероятно, это связано с make_axes_locatable и dentude_layout = true , но в моем случае у меня нет выбора использовать его.
Более того, когда я его сохраняю, я получаю весь сюжет, но соотношение аспекта не сохраняется (5,7). PrettyPrint-Override ">import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.axes_grid1.axes_divider import make_axes_locatable

def plot2d(data, ax):
im = ax.imshow(data, aspect="auto")

divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)
plt.colorbar(im, cax=cax, label="DATA")

fig = plt.figure(constrained_layout=True)
fig.set_size_inches(5, 7)
ax = fig.subplot_mosaic([["D1"]])

plot2d(np.random.rand(10, 1000), ax["D1"])
plt.savefig("foo3.pdf", bbox_inches="tight", dpi=100)

plt.show()


Подробнее здесь: https://stackoverflow.com/questions/783 ... -locatable