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,
