def make_plot(run_name, N): time = dt.datetime(2010, 01, 28, 1, 0) hvel, salt, turb = load_data(run_name) dia = tp.stackTransectPlotDC(cmap='Spectral_r') # dia.addSample('velocity', hvel, time, unit='m s-1', clabel='Velocity') dia.addSample('salt', salt, time, N=30, unit='psu', clabel='Salinity', clim=[0, 32]) dia.addSample('turb', turb, time, N=30, unit='kg m-3', clabel='Sediment', clim=[0, 1], cmap='YlOrBr') for ax in dia.axGrid: mt.addVelocityQuivers(ax, hvel, N, vSkip=3, hSkip=1, qcolor='k') ax.set_xticklabels(['%s' % d for d in range(20, 90, 10)]) dia.addTitle('') dia.showColorBar() plt.gcf().savefig('test_%d.png' % N)
def make_plot(detide_files): fig = plt.figure() dia = tsp.stackTransectPlotDC() for f in detide_files: data = dc.dataContainer.loadFromNetCDF(f) var = data.fieldNames[0] dia.addSample(var, data, 0, clabel=var, unit='units') dia.showColorBar() plt.gcf().savefig('hvel_salt_trcr_detide.png')