old_contents: import numpy as np
  import matplotlib.pyplot as plt
  import matplotlib.colors as colors
  import xgcm
  import cartopy.crs as ccrs
  
  from xmitgcm import open_mdsdataset 
  from matplotlib.mlab import bivariate_normal
  from cartopy.mpl.gridliner import LONGITUDE_FORMATTER, LATITUDE_FORMATTER
  
  dir0 = '/homedata/bderembl/runmit/test_southatlgyre3'
  
  ds0 = open_mdsdataset(dir0,iters='all',prefix=['U','V'])
  
  grid = xgcm.Grid(ds0)
  print(grid)
  
  Energie_cinetique_X = grid.interp((ds0.U.where(ds0.hFacW>0)*ds0.hFacW)**2, 'X') 
  Energie_cinetique_Y = grid.interp((ds0.V.where(ds0.hFacS>0)*ds0.hFacS)**2, 'Y')
  Energie_cinetique = Energie_cinetique_X + Energie_cinetique_Y
  print('Energie_cinetique')
  print(Energie_cinetique)
  
  
  i = 0
  nz = 0
  
  while (i < 150):
      i=i+1
      print(i)
      plt.figure(1)
      ax = plt.subplot(projection=ccrs.PlateCarree());
      Energie_cinetique[i,nz,:,:].plot.pcolormesh('XC', 'YC', ax=ax,vmin=0,norm=colors.PowerNorm(gamma=1./2.),vmax=10,cmap='ocean');
      plt.title('Kinetic Energy m²/s²')
      plt.text(5,5,i,ha='center',wrap=True)
      ax.coastlines()
      gl = ax.gridlines(draw_labels=True, alpha = 0.5, linestyle='--');
      gl.xlabels_top = False
      gl.ylabels_right = False
      gl.xformatter = LONGITUDE_FORMATTER
      gl.yformatter = LATITUDE_FORMATTER
      if (i < 10):
          plt.savefig('Kinetic_Energy-'+'00'+str(i)+'.png')
          plt.clf()
      elif (i > 9) and (i < 100):
          plt.savefig('Kinetic_Energy-'+'0'+str(i)+'.png')
          plt.clf()
      else:
          plt.savefig('Kinetic_Energy-'+str(i)+'.png')
          plt.clf()
  
new_contents: import numpy as np
  import matplotlib.pyplot as plt
  import matplotlib.colors as colors
  import xgcm
  import cartopy.crs as ccrs
  
  from xmitgcm import open_mdsdataset 
  from matplotlib.mlab import bivariate_normal
  from cartopy.mpl.gridliner import LONGITUDE_FORMATTER, LATITUDE_FORMATTER
  
  dir0 = '/homedata/bderembl/runmit/test_southatlgyre3'
  
  ds0 = open_mdsdataset(dir0,iters='all',prefix=['U','V'])
  
  grid = xgcm.Grid(ds0)
  print(grid)
  
  Kinetic_energy_X = grid.interp((ds0.U.where(ds0.hFacW>0)*ds0.hFacW)**2, 'X') 
  Kinetic_energy_Y = grid.interp((ds0.V.where(ds0.hFacS>0)*ds0.hFacS)**2, 'Y')
  Kinetic_energy = Kinetic_energy_X + Kinetic_energy_Y
  print('Kinetic_energy')
  print(Kinetic_energy)
  
  i = 0
  nz = 0
  
  while (i < 150):
      i=i+1
      print(i)
      plt.figure(1)
      ax = plt.subplot(projection=ccrs.PlateCarree());
      Energie_cinetique[i,nz,:,:].plot.pcolormesh('XC', 'YC', ax=ax,vmin=0,norm=colors.PowerNorm(gamma=1./2.),vmax=10,cmap='ocean');
      plt.title('Kinetic Energy m²/s²')
      plt.text(5,5,i,ha='center',wrap=True)
      ax.coastlines()
      gl = ax.gridlines(draw_labels=True, alpha = 0.5, linestyle='--');
      gl.xlabels_top = False
      gl.ylabels_right = False
      gl.xformatter = LONGITUDE_FORMATTER
      gl.yformatter = LATITUDE_FORMATTER
      if (i < 10):
          plt.savefig('Kinetic_Energy-'+'00'+str(i)+'.png')
          plt.clf()
      elif (i > 9) and (i < 100):
          plt.savefig('Kinetic_Energy-'+'0'+str(i)+'.png')
          plt.clf()
      else:
          plt.savefig('Kinetic_Energy-'+str(i)+'.png')
          plt.clf()
  
subject: Create 1_Energie_cinetique.py