一、概括

前面部分中，我們已經(jīng)對 WRF 的安裝和基礎(chǔ)運(yùn)行進(jìn)行了介紹，本次內(nèi)容分享主要一下 WRF 模擬中所使用的網(wǎng)格嵌套和土地類型如何使用 python 進(jìn)行繪制，示例程序和代碼可以后臺回復(fù)WRF的基礎(chǔ)要素繪制獲取。

二、土地類型

2.1 結(jié)果

2.2 主程序

# 土地類型
from netCDF4 import Dataset
import matplotlib.pyplot as plt
from matplotlib.cm import get_cmap
from mpl_toolkits.basemap import Basemap
from landuse_colormap import *

from wrf import to_np, getvar, smooth2d, get_basemap, latlon_coords

path=r'D:\Project\wrfout_d01_2023-07-01_06_00_00'
ncfile = Dataset(path)

LU_INDEX = getvar(ncfile, "LU_INDEX")

lats, lons = latlon_coords(LU_INDEX)

# Get the basemap object
bm = get_basemap(LU_INDEX)

# Create a figure
fig = plt.figure(figsize=(12,9))

# Add geographic outlines
bm.drawcoastlines(linewidth=0.25)
bm.drawstates(linewidth=0.25)
bm.drawcountries(linewidth=0.25)

# Convert the lats and lons to x and y.  Make sure you convert the lats and
# lons to numpy arrays via to_np, or basemap crashes with an undefined
# RuntimeError.
x, y = bm(to_np(lons), to_np(lats))

cm,labels = LU_MODIS21()
im = plt.pcolormesh(x,y,to_np(LU_INDEX),cmap=cm,vmax=len(labels)+1,vmin=1)
cbar = plt.colorbar(orientation='vertical',shrink=.98,pad=.02)
cbar.set_ticks(np.arange(1.5,len(labels)+1))
cbar.ax.set_yticklabels(labels)
plt.title('MODIS Land Use Categories')
plt.savefig('1.png',bbox_inches='tight',dpi=300)

三、網(wǎng)格嵌套

3.1 結(jié)果

3.2 主程序

from salem import geogrid_simulator
import matplotlib.gridspec as gridspec
import matplotlib.pyplot as plt

namelist_path = r'D:\Project\namelist.wps' # ensure no comments in namelist
g, maps = geogrid_simulator(namelist_path)
print(maps)
fig = plt.figure(1, figsize=(5, 5))
gs = gridspec.GridSpec(1, 1)
ax = fig.add_subplot(gs[0])

# 添加自定義shape文件
# maps[0].set_shapefile(r"/home/province_9south.shp")

# 繪制所有domain
maps[0].set_rgb(natural_earth='lr')
maps[0].visualize(title='Domains')
'''
# 單獨(dú)繪制d02
maps[1].set_rgb(natural_earth='lr') 
maps[1].visualize(title='Domains')
'''
gs.tight_layout(fig)
plt.show()