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] y = F.y[::n, ::n] F.bmap.quiver(x, y, U_diff2, V_diff2, scale=500) windmag = N.sqrt(U_diff ** 2 + V_diff ** 2) clvs = N.arange(5, 32.5, 2.5) cb = F.bmap.contourf(F.x, F.y, windmag, alpha=0.5, cmap=M.cm.copper_r, levels=clvs, extend="max") # fig.subplots_adjust(bottom=0.12,right=0.9) # cbar_ax = fig.add_axes([0.15,0.087,0.7,0.025]) # cbar_ax = False
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,) n = 3 U_diff2 = -1* U_diff[::n,::n] V_diff2 = -1* V_diff[::n,::n] x = F.x[::n,::n] y = F.y[::n,::n] F.bmap.quiver(x,y,U_diff2,V_diff2,scale=500) windmag = N.sqrt(U_diff**2 + V_diff**2) clvs = N.arange(5,32.5,2.5) cb = F.bmap.contourf(F.x,F.y,windmag,alpha=0.5,cmap=M.cm.copper_r,levels=clvs,extend='max') # F.bmap.plot((lonA,lonB),(latA,latB),latlon=True,color='k') # fig.subplots_adjust(bottom=0.12,right=0.9) # cbar_ax = fig.add_axes([0.15,0.087,0.7,0.025])
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) n = 3 U_diff2 = -1 * U_diff[::n, ::n] V_diff2 = -1 * V_diff[::n, ::n] x = F.x[::n, ::n] y = F.y[::n, ::n] F.bmap.quiver(x, y, U_diff2, V_diff2, scale=500) windmag = N.sqrt(U_diff ** 2 + V_diff ** 2) clvs = N.arange(5, 32.5, 2.5) cb = F.bmap.contourf(F.x, F.y, windmag, alpha=0.5, cmap=M.cm.copper_r, levels=clvs, extend="max") # F.bmap.plot((lonA,lonB),(latA,latB),latlon=True,color='k') # fig.subplots_adjust(bottom=0.12,right=0.9) # cbar_ax = fig.add_axes([0.15,0.087,0.7,0.025])