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') # ---------------------------------------------------------------------- # Zonal mean zonal wind season = 'jjas' lon1, lon2 = 60, 100 cint = 5 months = utils.season_months(season)
v = ds['v'] T = ds['T'] ps = ds['ps'] lat = get_coord(u, 'lat') lon = get_coord(u, 'lon') plev = get_coord(u, 'plev') mon = ds['mon'].values topo = dat.get_ps_clim(lat, lon) / 100 topo.units = 'hPa' # ---------------------------------------------------------------------- # Correct for topography u_orig = u u = dat.correct_for_topography(u_orig, topo) # ---------------------------------------------------------------------- # Integrated vertically dp/g # DataArray u_int = int_pres(u, pdim=-3) # ndarray u_int2 = int_pres(u.values, plev * 100, pdim=-3) p0 = 1e5 g = constants.g.values scale = g / p0 m = 7
v = ds['v'] T = ds['T'] ps = ds['ps'] lat = get_coord(u, 'lat') lon = get_coord(u, 'lon') plev = get_coord(u, 'plev') mon = ds['mon'].values topo = dat.get_ps_clim(lat, lon) / 100 topo.units = 'hPa' # ---------------------------------------------------------------------- # Correct for topography u_orig = u u = dat.correct_for_topography(u_orig, topo) # ---------------------------------------------------------------------- # Integrated vertically dp/g # DataArray u_int = int_pres(u, pdim=-3) # ndarray u_int2 = int_pres(u.values, plev*100, pdim=-3) p0=1e5 g = constants.g.values scale = g/p0 m = 7
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') # ---------------------------------------------------------------------- # Zonal mean zonal wind season='jjas' lon1, lon2 = 60, 100 cint = 5 months = utils.season_months(season)