Squeeze output in get_slice_miss only if more than 1 dimension

mcuntz · Jan 14, 2024 · edeeb30 · edeeb30
1 parent 5aa90a5
commit edeeb30
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/ncvue/ncvmap.py b/src/ncvue/ncvmap.py
@@ -32,7 +32,6 @@
     * Allow groups in netcdf files, Jan 2024, Matthias Cuntz
 
 """
-from __future__ import absolute_import, division, print_function
 import sys
 import tkinter as tk
 try:

diff --git a/src/ncvue/ncvmethods.py b/src/ncvue/ncvmethods.py
@@ -62,6 +62,8 @@
       (get_dim_var) Oct 2021, Matthias Cuntz
     * Address fi.variables[name] directly by fi[name], Jan 2024, Matthias Cuntz
     * Allow groups in netcdf files, Jan 2024, Matthias Cuntz
+    * Squeeze output in get_slice_miss only if more than 1 dim,
+      Jan 2024, Matthias Cuntz
 
 """
 from __future__ import absolute_import, division, print_function
@@ -557,8 +559,10 @@ def get_slice_miss(self, dimspins, x):
 
     """
     miss = get_miss(self, x)
-    xx   = get_slice(dimspins, x).squeeze()
-    xx   = set_miss(miss, xx)
+    xx = get_slice(dimspins, x)
+    if xx.ndim > 1:
+        xx = xx.squeeze()
+    xx = set_miss(miss, xx)
     # catch variables that have only one string or similar
     try:
         sx = xx.shape[0]