diff --git a/video_prediction_tools/postprocess/plot_ambs_forecast.py b/video_prediction_tools/postprocess/plot_ambs_forecast.py index 1f7da969d9b643663142dd6aaa18502235668eab..69ac8bf2b95579785bbd6e82b53c66e102849885 100644 --- a/video_prediction_tools/postprocess/plot_ambs_forecast.py +++ b/video_prediction_tools/postprocess/plot_ambs_forecast.py @@ -85,10 +85,10 @@ def create_plot(data: xr.DataArray, data_ref: xr.DataArray, varname: str, fcst_h m.drawparallels(np.arange(0, 90, 5),labels=lat_lab, xoffset=0.5) m.drawmeridians(np.arange(5, 355, 10),labels=lon_lab, yoffset=0.5) if t == 0: - cs = m.contourf(x, y, data.isel(time_forecast=fcst_hour)-273.15, clevs, cmap=plt.get_cmap("jet"), ax=axes[t]) + cs = m.contourf(x, y, data.isel(fcst_hour=fcst_hour)-273.15, clevs, cmap=plt.get_cmap("jet"), ax=axes[t]) cbar_ticks = None elif t == 1: - cs = m.contourf(x, y, data.isel(time_forecast=fcst_hour)-data_ref.isel(time_forecast=fcst_hour), + cs = m.contourf(x, y, data.isel(fcst_hour=fcst_hour)-data_ref.isel(fcst_hour=fcst_hour), clevs_diff, cmap=plt.get_cmap("PuOr"), ax=axes[t]) cbar_ticks = list(np.arange(-10.5, -2., 2.)) + [-0.5, 0.5] + list(np.arange(2.5, 10.6, 2.)) print(cbar_ticks)