def plot_coast_line(grd, proj=None): ''' plot_coast_line(grd, {proj}) plot the coastline from the object grid. proj=map (optional) is a Basemap object for projection. ''' a = plt.gca() coast = get_coast_line(grd) c = np.array(coast) if proj is None: col = collections.LineCollection(c) else: cp = np.zeros(c.shape) for i in range(c.shape[0]): cp[i,:,0], cp[i,:,1] = proj(c[i,:,0], c[i,:,1]) col = collections.LineCollection(cp) a.add_collection(col, autolim=True) col.set_color('k')
def plot_coast_line(grd, proj=None): ''' plot_coast_line(grd, {proj}) plot the coastline from the object grid. proj=map (optional) is a Basemap object for projection. ''' a = plt.gca() coast = get_coast_line(grd) c = np.array(coast) if proj is None: col = collections.LineCollection(c) else: cp = np.zeros(c.shape) for i in range(c.shape[0]): cp[i, :, 0], cp[i, :, 1] = proj(c[i, :, 0], c[i, :, 1]) col = collections.LineCollection(cp) a.add_collection(col, autolim=True) col.set_color('k')
lon = grd.hgrid.lon_rho lat = grd.hgrid.lat_rho h = grd.vgrid.h mask = grd.hgrid.mask_rho idx = np.where(mask == 0) h[idx] = 0 fig = plt.figure(figsize=(10, 5)) ax = fig.gca(projection='3d', azim=-50, elev=70) pal = cm.spectral pal.set_over('#666666', 1.0) vc = np.arange(-80,-5+0.1,0.1) pal_norm = colors.BoundaryNorm(vc,ncolors=256, clip = False) surf = ax.plot_surface(lon, lat, -h, rstride=1, cstride=1, cmap=pal, norm=pal_norm, linewidth=0, antialiased=False) fig.colorbar(surf, shrink=0.9, aspect=40) coast = pyroms_toolbox.get_coast_line(grd) coast = np.array(coast) coast = collections.LineCollection(coast) line_collection_2d_to_3d(coast, zs=-1, zdir='z') coast.set_color('k') ax.add_collection3d(coast, zs=-1, zdir='z') outfile='YELLOW_bathy_3D.png' plt.savefig(outfile, dpi=300, facecolor='w', edgecolor='w', \ orientation='portrait')
lon = grd.hgrid.lon_rho lat = grd.hgrid.lat_rho h = grd.vgrid.h mask = grd.hgrid.mask_rho idx = np.where(mask == 0) h[idx] = 0 fig = plt.figure(figsize=(10, 5)) ax = fig.gca(projection='3d', azim=-50, elev=70) pal = cm.Spectral pal.set_over('#666666', 1.0) vc = np.arange(-80,-5+0.1,0.1) pal_norm = colors.BoundaryNorm(vc,ncolors=256, clip = False) surf = ax.plot_surface(lon, lat, -h, rstride=1, cstride=1, cmap=pal, norm=pal_norm, linewidth=0, antialiased=False) fig.colorbar(surf, shrink=0.9, aspect=40) coast = pyroms_toolbox.get_coast_line(grd) coast = np.array(coast) coast = collections.LineCollection(coast) line_collection_2d_to_3d(coast, zs=-1, zdir='z') coast.set_color('k') ax.add_collection3d(coast, zs=-1, zdir='z') outfile='YELLOW_bathy_3D.png' plt.savefig(outfile, dpi=300, facecolor='w', edgecolor='w', \ orientation='portrait')