Analysis of experiments

experiments/ANN-Results/helpers_mom6/collection_of_experiments.py

@@ -239,9 +239,9 @@ def plot_ssh(self, exps, labels=None, target=None, ncols=3):
                 label = 'SSH error [m]'
                 lines = False
             ssh.plot.contourf(levels=levels, cmap='bwr', linewidths=1, extend='both', cbar_kwargs={'label': label})
-            if lines:
-                Cplot = ssh.plot.contour(levels=levels, colors='k', linewidths=1)
-                plt.gca().clabel(Cplot, Cplot.levels)
+            #if lines:
+            #     Cplot = ssh.plot.contour(levels=levels, colors='k', linewidths=1)
+            #     plt.gca().clabel(Cplot, Cplot.levels)
             plt.xticks((0, 5, 10, 15, 20))
             plt.yticks((30, 35, 40, 45, 50))
             plt.xlabel('Longitude')
@@ -255,6 +255,41 @@ def plot_ssh(self, exps, labels=None, target=None, ncols=3):
 
         plt.tight_layout()
 
+    def plot_SGSmean(self, exps, labels=None, target=None, ncols=3, zl=0):
+        if labels is None:
+            labels=exps
+        nfig = len(exps)
+        ncol = min(ncols,nfig)
+        nrows = nfig / ncols
+        if nrows > 1:
+            nrows = int(np.ceil(nrows))
+        else:
+            nrows = 1
+
+        plt.figure(figsize=(5*ncol,4*nrows))
+        plt.subplots_adjust(hspace=0.3, wspace=0.3)
+
+        for ifig, exp in enumerate(exps):
+            plt.subplot(nrows,ncol,ifig+1)
+            try:
+                data = self[exp].SGS_mean.isel(zl=zl) * 1e+7
+
+                cmap = plt.cm.seismic
+                vmin,vmax,ci = -1.45,1.45,0.1
+                cilev = np.arange(vmin,vmax+ci,ci)
+                norm = plt.matplotlib.colors.BoundaryNorm(boundaries=cilev, ncolors=cmap.N)
+
+                label = 'Time-mean zonal acceleration\n by subgrid eddies, $10^{-7}\mathrm{m}\mathrm{s}^{-2}$'
+                data.plot.pcolormesh(cmap=cmap, norm=norm, cbar_kwargs={'label': label})
+                plt.xticks((0, 5, 10, 15, 20))
+                plt.yticks((30, 35, 40, 45, 50))
+                plt.xlabel('Longitude')
+                plt.ylabel('Latitude')
+                plt.title(labels[ifig])
+            except:
+                pass
+        plt.tight_layout()
+
     def plot_ssh_std(self, exps, labels=None, target='R64_R2', ncols=3):
         if labels is None:
             labels=exps

experiments/ANN-Results/helpers_mom6/experiment.py

@@ -243,6 +243,10 @@ def h_mean(self):
     @netcdf_property
     def e_mean(self):
         return self.ea.sel(Time=self.Averaging_time).mean(dim='Time')
+
+    @netcdf_property
+    def SGS_mean(self):
+        return self.mom.ZB2020u.sel(Time=self.Averaging_time).mean(dim='Time')
 
     def average(self, prop):
         return eval(f'self.{prop}').sel(Time=self.Averaging_time).mean(dim='Time')

experiments/ANN-Results/offline_analysis/10-ANN-32-32.ipynb

@@ -2150,7 +2150,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.11.11"
   }
  },
  "nbformat": 4,