用python怎么實(shí)現(xiàn)多子圖共享色標(biāo)

用EAR5 hourly data on pressure levels from 1979 to present的再分析資料，繪制了200hPa、500hPa、700hPa和850hPa的相對濕度和風(fēng)場。

首先，我們要新建一個大的畫布，設(shè)置好它的大小和分辨率：

fig = plt.figure(figsize=(12,8),dpi=150)

我們需要畫同一時刻四個不同層次，因此我們在這個大畫布里設(shè)置四個子圖，給它們各自起一個名字：

ax1 = plt.subplot(221, projection = proj)
ax2 = plt.subplot(222, projection = proj)
ax3 = plt.subplot(223, projection = proj)
ax4 = plt.subplot(224, projection = proj)

（這里沒有用subplots()是因為它好像不支持ccrs.PlateCarree()的投影方式，我就只能這樣來做了。而且不能用這個新建子圖的方式也讓我沒辦法令子圖共用坐標(biāo)軸，不知道具體是哪里出了問題。。。）

但是，在調(diào)用函數(shù)的時候，當(dāng)然還是名字都一樣好寫一點(diǎn)咯~只畫4層情況的話，我使用了range來控制讀取的層次，那么，如何將有規(guī)律的名字字符串轉(zhuǎn)換為變量的名字，再傳入自定義函數(shù)中呢？我使用了locals()來達(dá)成目的。

for lev in range(0,4):
    r = obj['r'][0][lev][:][:]
    u = obj['u'][0][lev][:][:]
    v = obj['v'][0][lev][:][:]

    the_string = f'ax{lev+1}'
    ax = locals()[the_string]
       
    ac = RHLevel(r, u, v, ax, lev)

為什么要令函數(shù)等于“ac”呢？由于四張圖片所使用的填色規(guī)則都是一樣的，為了看起來順眼，只需要一條colorbar作為圖例會好很多。所以，我們可以將畫相對濕度的語句傳出，在循環(huán)的外部利用該信息繪制整張圖的colorbar。

def RHLevel(r, u, v, ax, lev):
    ......
    ac = ax.contourf(lons[248:288,420:480],lats[248:288,420:480],r[248:288,420:480],levels=np.arange(60,110,10),
               cmap='Greens',extend='both', alpha=0.9)
    return ac

由于返回了ac的信息，所以在調(diào)用函數(shù)之后該函數(shù)是有一個“結(jié)果”的，可以為這個結(jié)果“起名”，這樣就可以在循環(huán)外用傳遞出來的結(jié)果來自定義colorbar了：

ac = RHLevel(r, u, v, ax, lev)
cb = fig.colorbar(ac, cax = cbar_ax,orientation='horizontal',spacing='proportional')

然后，就可以利用colorbar指令來定義它的位置、方向了，在這里，l, b, w, h分別是指該colorbar的左側(cè)起始點(diǎn)、下側(cè)起始點(diǎn)、寬度和高度，數(shù)字則是百分比，比如說0.25是指大畫布的寬的0.25倍：

l,b,w,h = 0.25, 0.05, 0.5, 0.02
rect = [l,b,w,h]
cbar_ax = fig.add_axes(rect)
cb = fig.colorbar(ac, cax = cbar_ax,orientation='horizontal',spacing='proportional')

為了標(biāo)出不同圖片所對應(yīng)的層次，我們可以在函數(shù)中就給每一個子圖設(shè)置一個小標(biāo)題。loc=''可以設(shè)置該標(biāo)題的對齊位置，在這里標(biāo)題向左對齊。

    ax.set_title(f'{level[lev]}hPa',fontsize=12,loc='left')

在大圖（循環(huán)外）中設(shè)置整張圖的標(biāo)題（和子圖標(biāo)題不一樣噢），具體有什么區(qū)別可以搜一下fig/ax/plt的區(qū)別，有不少文章都講得很詳細(xì)：

plt.suptitle(f'2020年5月21日00時（世界時）各層相對濕度與風(fēng)場',fontsize=18)