from wrf import uvmet, to_np, getvar, interplevel, smooth2d, get_cartopy, cartopy_xlim, cartopy_ylim, latlon_coords
import numpy as np
from netCDF4 import Dataset
import xarray as xr
from metpy.units import units
import matplotlib.pyplot as plt
from matplotlib.cm import get_cmap
from matplotlib.colors import from_levels_and_colors
import cartopy.crs as ccrs
import cartopy.feature as cfeature
import cartopy.io.shapereader as shpreader
from cartopy.feature import NaturalEarthFeature
from cartopy.mpl.ticker import LongitudeFormatter, LatitudeFormatter
from cartopy.mpl.gridliner import LATITUDE_FORMATTER, LONGITUDE_FORMATTER
import cmaps
from glob import glob
import metpy.calc as mpcalc
import metpy.constants as constants

Warning: ecCodes 2.21.0 or higher is recommended. You are running version 2.14.1

# использовать netCDF4 Чтение пакетов WRF моделированиеданные
wrf_file = Dataset('/home/mw/input/wrfout3385/wrfout_d02_2022-07-14_0600.nc')
# Извлечение функций
lon = getvar(wrf_file, 'lon', timeidx=0)
lat = getvar(wrf_file, 'lat', timeidx=0)
u = getvar(wrf_file, 'ua', timeidx=0)
v = getvar(wrf_file, 'va', timeidx=0)
q = getvar(wrf_file, 'QVAPOR', timeidx=0)*1000
# Извлеките долготу и широту моделирования WRF из множества
lats, lons = latlon_coords(u)

# Извлечение картографической проекции моделирования WRF
wrf_proj = get_cartopy(u)

p = getvar(wrf_file, 'pressure', timeidx=0)
u850 = interplevel(u, p, 850)
v850 = interplevel(v, p, 850)
q850 = interplevel(q, p, 850) ** units('g/kg')
z = getvar(wrf_file, 'z', units="dm", timeidx=0)
z850 = interplevel(z, p, 850)
dx, dy = mpcalc.lat_lon_grid_deltas(lon,lat)
# вычислитьрасходимость потока водяного пара
q_flux_divergence = mpcalc.divergence(
    to_np(
        u850 *
        q850/constants.g),
    to_np(
        v850 *
        q850/constants.g),
    dx=to_np(dx),
    dy=to_np(dy))

u_interp = interplevel(u, p, levels)
v_interp = interplevel(v, p, levels)
q_interp = interplevel(q, p,уровни) ** единицы('г/кг')
лев = u_interp.level

# Спасибо, что прочитали эту страницу
qu = u_interp *q_interp/constants.g
qv = v_interp *q_interp/constants.g
q_flux_divergence_all = np.zeros((lev.shape[0],lat.shape[0],lon.shape[0]))
for i in range(lev.shape[0]):
    q_flux_divergence_all[i] = mpcalc.divergence(u = to_np(qu[i]),v = to_np(qv[i]),dx = to_np(dx) ,dy = to_np(dy),x_dim=-1, y_dim=-2)

# Воля q_flux_divergence_all в NaN Значение заменяется на 0
q_flux_divergence_all = np.nan_to_num(q_flux_divergence_all, nan=0)
total_div_qv = np.trapz(q_flux_divergence_all,lev,axis=0)
total_div_qv[2]