python - Colorbars close to subplots -
how can put colorbars beside each colormap in subplot? simplified version of real code here shows problem. can see colorbars @ bottom right , making last plot smaller.
from matplotlib import pyplot plt import numpy np def plots(): fig,ax=plt.subplots(2,2) in range(2): rho_mat,c_mat=np.random.uniform(size=(50,50)),np.random.uniform(size=(50,50)) ax[0,i].set_title(r"$\rho_{x,y}$") p=ax[0,i].imshow(np.fliplr(rho_mat).t,extent=[0.,1,0.,1],vmin=0,vmax=1, interpolation='none') ax[0,i].set_xlabel(r'$\epsilon_1$') ax[0,i].set_ylabel(r'$\epsilon_2$') fig.colorbar(p, shrink=0.5) ax[1,i].set_title(r"$c_{x,y}$") p2=ax[1,i].imshow(np.fliplr(c_mat).t,extent=[0.,1,0.,1],vmin=0,vmax=1, interpolation='none') ax[1,i].set_xlabel(r'$\epsilon_1$') ax[1,i].set_ylabel(r'$\epsilon_2$') fig.colorbar(p2, shrink=0.5) plt.tight_layout() plt.show() plots()
the colorbar method has optional keyword argument allows specify axes associated with.
in code change call colorbar this:
fig.colorbar(p, ax=ax[0,i], shrink=0.5) fig.colorbar(p2, ax=ax[1,i], shrink=0.5)
Comments
Post a Comment