Open the input file only once

Inst_eddies/Analysis/plot_velocity.py

@@ -33,6 +33,13 @@ with netCDF4.Dataset(args.input_file) as f:
     longitude = f[lon][:]
     latitude = f[lat][:]
 
+    if "time" in f["ugos"].dimensions:
+        ugos = f["ugos"][0]
+        vgos = f["vgos"][0]
+    else:
+        ugos = f["ugos"][:]
+        vgos = f["vgos"][:]
+
 if args.window is None:
     lon_mask = np.ones(len(longitude), dtype = bool)
     lat_mask = np.ones(len(latitude), dtype = bool)
@@ -55,14 +62,6 @@ projection = ccrs.PlateCarree()
 fig = plt.figure()
 ax = plt.axes(projection = projection)
 
-with netCDF4.Dataset(args.input_file) as f:
-    if "time" in f["ugos"].dimensions:
-        ugos = f["ugos"][0]
-        vgos = f["vgos"][0]
-    else:
-        ugos = f["ugos"][:]
-        vgos = f["vgos"][:]
-
 if args.undefined:
     undef_velocity = np.logical_or(ugos.mask, vgos.mask)
     lon_2d, lat_2d = np.meshgrid(longitude[lon_mask],