Commit 87ded12a authored by Modellers Operational's avatar Modellers Operational

Bug fix to allow plotting of zeta

parent a1597e04
......@@ -32,7 +32,10 @@ labels = {'q2': 'Turbulent kinetic energy $(m^{2}s^{-2})$',
def plot_var(idx):
plot = pf.plot.Plotter(fvcom, figsize=(23, 18), cmap=cmap, cb_label=label, extend=extension, res=None)
plot.plot_field(np.squeeze(getattr(fvcom.data, var))[idx, level, :])
if var == 'zeta':
plot.plot_field(np.squeeze(getattr(fvcom.data, var))[idx, :])
else:
plot.plot_field(np.squeeze(getattr(fvcom.data, var))[idx, level, :])
plot.tripcolor_plot.set_clim(clim[0], clim[1])
plot.axes.set_title(fvcom.time.Times[idx][:-7].replace('T', ' '))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment