Как иметь одну цветовую панель для всех подграфиков ⇐ Python

[Anonymous]

Я потратил слишком много времени на исследование того, как заставить два подграфика использовать одну и ту же ось Y с одной цветовой полосой, общей для них в Matplotlib.

Произошло следующее: когда я вызывал функцию colorbar() либо в subplot1, либо в subplot2, она автоматически масштабировала график так, что цветная полоса и график поместятся внутри ограничивающего прямоугольника «подграфик», в результате чего два расположенных рядом графика будут иметь совершенно разные размеры.

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

Вот мой код:
из отдела импорта __future__ импортировать matplotlib.pyplot как plt импортировать numpy как np из патчей импорта matplotlib из matplotlib.ticker импортировать NullFormatter # Функции СПАЗ TE = 1 # радиус Эйнштейна g1 = лямбда x,y: (TE/2) * (y**2-x**2)/((x**2+y**2)**(3/2)) g2 = лямбда x,y: -1*TE*x*y / ((x**2+y**2)**(3/2)) каппа = лямбда x,y: TE / (2*np.sqrt(x**2+y**2)) координаты = np.linspace(-2,2,400) X,Y = np.meshgrid(координаты,координаты) g1out = g1(X,Y) g2out = g2(X,Y) каппааут = каппа(X,Y) для i в диапазоне (len(coords)): для j в диапазоне (len(coords)): если np.sqrt(coords**2+coords[j]**2)