def get_subdomain(self,Nlim,Elim,Slim,Wlim): """ Return data array between bounds """ data,lats,lons = utils.return_subdomain(self.data,self.lats, self.lons,Nlim,Elim,Slim,Wlim) return data,lats,lons
def get_subdomain(self,Nlim,Elim,Slim,Wlim,overwrite=False): """ Return data array between bounds If overwrite is True, replace class data with new subdomain """ data,lats,lons = utils.return_subdomain(self.data,self.lats, self.lons,Nlim,Elim,Slim,Wlim) if overwrite: self.lats = lats self.lons = lons self.data = data return else: return data,lats,lons
def get_subdomain(self, Nlim, Elim, Slim, Wlim, overwrite=False): """ Return data array between bounds If overwrite is True, replace class data with new subdomain """ data, lats, lons = utils.return_subdomain(self.data, self.lats, self.lons, Nlim, Elim, Slim, Wlim) if overwrite: self.lats = lats self.lons = lons self.data = data return else: return data, lats, lons
DATA = {} # cross-section labels latA = 40.3 lonA = -99.3 latB = 39.0 lonB = -98.2 for nest in ("SINGLE","NESTED"): DATA[nest] = {} DATA[nest]['W'] = WRFOut(os.path.join(ncdir[nest],ncfile)) # import pdb; pdb.set_trace() DATA[nest]['U'] = DATA[nest]['W'].get('U',utc=utc,level=lv)[0,0,:,:] DATA[nest]['U'],lats,lons = utils.return_subdomain(DATA[nest]['U'], DATA[nest]['W'].lats1D,DATA[nest]['W'].lons1D, fmt='latlon',**lims) DATA[nest]['V'] = DATA[nest]['W'].get('V',utc=utc,level=lv)[0,0,:,:] DATA[nest]['V'],lats,lons = utils.return_subdomain(DATA[nest]['V'], DATA[nest]['W'].lats1D,DATA[nest]['W'].lons1D, fmt='latlon',**lims) U_diff = DATA['SINGLE']['U'] - DATA['NESTED']['U'] V_diff = DATA['SINGLE']['V'] - DATA['NESTED']['V'] # difference plot diffplot = 0 if diffplot: fig,ax = plt.subplots(1,figsize=(4,4)) F = BirdsEye(ax=ax,fig=fig) F.bmap,F.x,F.y = F.basemap_setup(lats=lats,lons=lons,)
utc = (2013, 8, 15, 21, 0, 0) outdir = "/home/jrlawson/public_html/bowecho/paper2" lims = {"Nlim": 41.0, "Elim": -97.4, "Slim": 38.3, "Wlim": -101.0} fig, ax = plt.subplots(1, figsize=(4, 4)) DATA = {} for nest in ("SINGLE", "NESTED"): DATA[nest] = {} DATA[nest]["W"] = WRFOut(os.path.join(ncdir[nest], ncfile)) import pdb pdb.set_trace() DATA[nest]["U"] = DATA[nest]["W"].get("U", utc=utc, level=lv)[0, 0, :, :] DATA[nest]["U"], lats, lons = utils.return_subdomain( DATA[nest]["U"], DATA[nest]["W"].lats1D, DATA[nest]["W"].lons1D, fmt="latlon", **lims ) DATA[nest]["V"] = DATA[nest]["W"].get("V", utc=utc, level=lv)[0, 0, :, :] DATA[nest]["V"], lats, lons = utils.return_subdomain( DATA[nest]["V"], DATA[nest]["W"].lats1D, DATA[nest]["W"].lons1D, fmt="latlon", **lims ) U_diff = DATA["SINGLE"]["U"] - DATA["NESTED"]["U"] V_diff = DATA["SINGLE"]["V"] - DATA["NESTED"]["V"] F = BirdsEye(ax=ax, fig=fig) F.bmap, F.x, F.y = F.basemap_setup(lats=lats, lons=lons) n = 3 U_diff2 = -1 * U_diff[::n, ::n] V_diff2 = -1 * V_diff[::n, ::n] x = F.x[::n, ::n]