Matplotlib 2 sottotrame, 1 barra dei colori

| | |

Ho passato troppo tempo a cercare come ottenere due sottotrame per condividere lo stesso asse y con una singola barra dei colori condivisa tra i due in Matplotlib.

Quello che stava succedendo è che quando ho chiamato il colorbar() in subplot1 o subplot2, ridimensionerebbe automaticamente il grafico in modo tale che la barra dei colori più il grafico rientrino nella "sottotrama" riquadro di delimitazione, facendo in modo che i due grafici affiancati abbiano due dimensioni molto diverse.

Per aggirare questo problema, ho provato a creare una terza sottotrama che ho poi modificato per non renderizzare alcun grafico con solo una barra dei colori presente. L'unico problema è che ora le altezze e le larghezze dei due grafici non sono uniformi e non riesco a capire come farlo sembrare a posto.

Ecco il mio codice:

from __future__ import division import matplotlib.pyplot as plt import numpy as np from matplotlib import patch from matplotlib.ticker import NullFormatter # SIS Functions TE = 1 # Einstein raggio g1 = lambda x,y: (TE/2) * (s**2 -x**2)/((x**2+y**2)**(3/2)) g2 = lambda x,y: -1*TE*x*y / ((x**2+ y**2)**(3/2)) kappa = lambda x,y: TE / (2*np.sqrt(x**2+y**2)) coords = np.linspace(-2,2,400 ) X,Y = np.meshgrid(coords,coords) g1out = g1(X,Y) g2out = g2(X,Y) kappaout = kappa(X,Y) for i in range(len(coords)): for j in range(len(coords)): if np.sqrt(coords[i]**2+coords[j]**2) <= TE: g1out[i][j]=0 g2out[i][j ]=0 fig = plt.figure() fig.subplots_adjust(wspace=0,hspace=0) # sottotrama numero 1 ax1 = fig.add_subplot(1,2,1,aspect="equal",xlim=[-2, 2],ylim=[-2,2]) plt.title(r"$gamma_{1}$",fontsize="18") plt.xlabel(r"x ($ heta_{E}$)",fontsize ="15") plt.ylabel(r"y ($ heta_{E}$)",rotation="horizontal",fontsize="15") plt.xticks([-2.0,-1.5,-1.0,-0.5 ,0,0.5,1.0,1.5]) plt.xticks([-2.0,-1.5,-1.0,-0.5,0,0.5,1.0,1.5]) plt.imshow(g1out,extent=(-2,2, -2,2)) plt.axhline(y=0,linewidth=2,color="k",linestyle="--") plt.axvline(x=0,linewidth=2,color="k",linestyle ="--") e1 = patch.Ellipse((0,0),2,2,color="white") ax1.add_patch(e1) # sottotrama numero 2 a x2 = fig.add_subplot(1,2,2,sharey=ax1,xlim=[-2,2],ylim=[-2,2]) plt.title(r"$gamma_{2}$",fontsize= "18") plt.xlabel(r"x ($ heta_{E}$)", fontsize="15") ax2.yaxis.set_major_formatter( NullFormatter() ) plt.axhline(y=0,linewidth=2,color ="k",linestyle="--") plt.axvline(x=0,linewidth=2,color="k",linestyle="--") plt.imshow(g2out,extent=(-2,2 ,-2,2)) e2 = patch.Ellipse((0,0),2,2,color="white") ax2.add_patch(e2) # sottotrama per la barra dei colori ax3 = fig.add_sottotrama(1,1,1 ) ax3.axis("off") cbar = plt.colorbar(ax=ax2) plt.show()