Matplotlib 2 個子圖,1 個顏色條

| | |

我花了很長時間研究如何讓兩個子圖共享同一個 y 軸,並在 Matplotlib 中兩個子圖之間共享一個顏色條。

發生的事情是,當我調用colorbar() 函數在 subplot1subplot2 中,它會自動縮放繪圖,使顏色條加上繪圖適合“子繪圖”邊界框,導致兩個並排的圖是兩個非常不同的大小。

為了解決這個問題,我嘗試創建第三個子圖,然後我破解了只用顏色條渲染沒有圖目前。唯一的問題是,現在兩個地塊的高度和寬度是不均勻的,我不知道如何讓它看起來不錯。

這是我的代碼:

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 radius g1 = lambda x,y: (TE/2) * (y**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在範圍內(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) # 子圖編號 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 = patchs.Ellipse((0,0),2,2,color="white") ax1.add_patch(e1) # subplot number 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) # 顏色條的子圖 ax3 = fig.add_subplot(1,1,1 ) ax3.axis("off") cbar = plt.colorbar(ax=ax2) plt.show()