# ---------------------------------------------------------------------- # Read monthly mean climatologies and do some test calcs # ---------------------------------------------------------------------- filename = 'data/more/ncep2_climatology_monthly.nc' ds = dat.ncload(filename) lat = ds['lat'].values lon = ds['lon'].values plev = ds['plev'].values mon = ds['mon'].values u = ds['u'] v = ds['v'] T = ds['T'] ps = ds['ps'] topo = get_ps_clim(lat, lon) topo.values /= 100 topo.attrs['units'] = 'hPa' # ---------------------------------------------------------------------- # Correct for topography u_orig = u u = correct_for_topography(u_orig, topo) m, k = 3, 1 plt.figure(figsize=(7, 8)) plt.subplot(211) ap.pcolor_latlon(u_orig[m, k], cmap='jet') plt.subplot(212) ap.pcolor_latlon(u[m, k], cmap='jet')
# ----------------------------------------------------------------------
# Read data
url = ('http://goldsmr3.sci.gsfc.nasa.gov/opendap/MERRA_MONTHLY/'
'MAIMCPASM.5.2.0/1979/MERRA100.prod.assim.instM_3d_asm_Cp.197907.hdf')
ds = xray.open_dataset(url)
#T = ds['T']
#ps = ds['PS']
u = ds['U']
v = ds['V']
q = ds['QV']
lat = get_coord(u, 'lat')
lon = get_coord(u, 'lon')
plev = get_coord(u, 'plev')
topo = dat.get_ps_clim(lat, lon) / 100
topo.units = 'hPa'
# ----------------------------------------------------------------------
# Initial plots
ubar = u.mean(axis=-1)
vbar = v.mean(axis=-1)
plt.figure(figsize=(7,8))
plt.subplot(211)
ap.contour_latpres(ubar, clev=5)
plt.subplot(212)
ap.contour_latpres(vbar, clev=0.5)
t, k = 0, 4
# ----------------------------------------------------------------------
# Read data
url = ('http://goldsmr3.sci.gsfc.nasa.gov/opendap/MERRA_MONTHLY/'
'MAIMCPASM.5.2.0/1979/MERRA100.prod.assim.instM_3d_asm_Cp.197907.hdf')
ds = xray.open_dataset(url)
#T = ds['T']
#ps = ds['PS']
u = ds['U']
v = ds['V']
#q = ds['QV']
lat = get_coord(v, 'lat')
lon = get_coord(v, 'lon')
plev = get_coord(v, 'plev')
topo = dat.get_ps_clim(lat, lon) / 100
topo.units = 'hPa'
# ----------------------------------------------------------------------
# Relative and absolute vorticity
# DataArray
rel_vort, abs_vort, f = vorticity(u, v)
# ndarray
rel_vort2, abs_vort2, f2 = vorticity(u.values, v.values, lat, lon)
t, k = 0, 22
plt.figure(figsize=(12,8))
plt.subplot(221)
# ---------------------------------------------------------------------- # Read monthly mean climatologies and do some test calcs # ---------------------------------------------------------------------- filename = 'data/more/ncep2_climatology_monthly.nc' ds = dat.ncload(filename) lat = ds['lat'].values lon = ds['lon'].values plev = ds['plev'].values mon = ds['mon'].values u = ds['u'] v = ds['v'] T = ds['T'] ps = ds['ps'] topo = get_ps_clim(lat, lon) topo.values /= 100 topo.attrs['units'] = 'hPa' # ---------------------------------------------------------------------- # Correct for topography u_orig = u u = correct_for_topography(u_orig, topo) m, k = 3, 1 plt.figure(figsize=(7,8)) plt.subplot(211) ap.pcolor_latlon(u_orig[m,k], cmap='jet') plt.subplot(212) ap.pcolor_latlon(u[m,k], cmap='jet')
