Changeset 3681

Timestamp:

Mar 12, 2025, 11:06:49 AM (3 months ago)

Author:

jbclement

Message:

PEM:
Small improvements for the python script to display variables in a NetCDF file.
JBC

Location:

trunk/LMDZ.MARS

Files:

• changelog.txt (modified) (1 diff)
• util/display_netcdf.py (modified) (3 diffs)

trunk/LMDZ.MARS/changelog.txt

@@ -4776 +4776 @@
 == 26/02/2025 == JBC
 Skipping non-numeric variables in the analysis of NetCDF files.
+
+== 12/03/2025 == JBC
+Small improvements for the python script to display variables in a NetCDF file.

trunk/LMDZ.MARS/util/display_netcdf.py

@@ -24 +24 @@
     # Add '/' if the match is a directory
     matches = [match + '/' if os.path.isdir(match) else match for match in matches]
-    
     try:
         return matches[state]
@@ -81 +80 @@
     # If the variable has a time dimension, ask for the time index
     if 'Time' in dimensions:
-        time_index = int(input(f"Enter the time index (0 to {variable.shape[0] - 1}): "))
+        if variable.shape[0] == 1:
+            time_index = 0
+        else:
+            time_index = int(input(f"Enter the time index (0 to {variable.shape[0] - 1}): "))
     else:
         time_index = None
@@ -103 +105 @@
     
     # Plot the selected variable
-    plt.figure(figsize=(10,6))
-    plt.contourf(longitude,latitude,data_slice,cmap='viridis')
+    plt.figure(figsize = (10,6))
+    plt.contourf(longitude,latitude,data_slice,cmap = 'jet')
     plt.colorbar(label=f"{variable_name.capitalize()} (units)") # Adjust units based on your data
     plt.xlabel('Longitude (degrees)')