Exemple #1
0
    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
Exemple #2
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,)
    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])