def testing(data, lat1, lat2, lon1, lon2, t, k): latname = get_coord(data, 'lat', 'name') lonname = get_coord(data, 'lon', 'name') data_sub = dat.subset(data, latname, lat1, lat2, lonname, lon1, lon2) plt.figure() ap.pcolor_latlon(data_sub[t, k], axlims=(lat1, lat2, lon1, lon2), cmap='jet') avg0 = data_sub.mean(axis=-1).mean(axis=-1) avg1 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=False) avg2 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=True) avg3 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=True, land_only=True) print(avg0[t, k].values) print(avg1[t, k].values) print(avg2[t, k].values) print(avg3[t, k].values)
def testing(data, lat1, lat2, lon1, lon2, t, k): latname = get_coord(data, 'lat', 'name') lonname = get_coord(data, 'lon', 'name') data_sub = dat.subset(data, latname, lat1, lat2, lonname, lon1, lon2) plt.figure() ap.pcolor_latlon(data_sub[t,k], axlims=(lat1,lat2,lon1,lon2), cmap='jet') avg0 = data_sub.mean(axis=-1).mean(axis=-1) avg1 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=False) avg2 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=True) avg3 = mean_over_geobox(data, lat1, lat2, lon1, lon2, area_wtd=True, land_only=True) print(avg0[t, k].values) print(avg1[t, k].values) print(avg2[t, k].values) print(avg3[t, k].values)
T = ds['T'] ps = ds['PS'] q = ds['QV'] lat = get_coord(ps, 'lat') lon = get_coord(ps, 'lon') plev = get_coord(T, 'plev') # ---------------------------------------------------------------------- # Averaging over box - constant array lon1, lon2 = 20, 80 lat1, lat2 = 65, 85 data = 2.5 * np.ones(ps.shape) avg1 = mean_over_geobox(data, lat1, lat2, lon1, lon2, lat=lat, lon=lon, area_wtd=False) avg2 = mean_over_geobox(data, lat1, lat2, lon1, lon2, lat=lat, lon=lon, area_wtd=True) avg3 = mean_over_geobox(data, lat1, lat2, lon1, lon2, lat=lat, lon=lon, area_wtd=True, land_only=True) print(avg1) print(avg2) print(avg3) # ---------------------------------------------------------------------- # Averaging over box def testing(data, lat1, lat2, lon1, lon2, t, k): latname = get_coord(data, 'lat', 'name') lonname = get_coord(data, 'lon', 'name')
lat = get_coord(ps, 'lat') lon = get_coord(ps, 'lon') plev = get_coord(T, 'plev') # ---------------------------------------------------------------------- # Averaging over box - constant array lon1, lon2 = 20, 80 lat1, lat2 = 65, 85 data = 2.5 * np.ones(ps.shape) avg1 = mean_over_geobox(data, lat1, lat2, lon1, lon2, lat=lat, lon=lon, area_wtd=False) avg2 = mean_over_geobox(data, lat1, lat2, lon1, lon2, lat=lat, lon=lon, area_wtd=True) avg3 = mean_over_geobox(data, lat1, lat2, lon1,