add squeeze array to MV2

CDAT · Nov 16, 2017 · 197aa46 · durack1 · Nov 16, 2017 · 197aa46
1 parent 46b27f5
commit 197aa46
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/Lib/MV2.py b/Lib/MV2.py
@@ -340,6 +340,14 @@ def _conv_axis_arg(axis):
         axis = 0
     return axis
 
+def squeeze(x):
+    "squeze array"
+    ta = _makeMaskedArg(x)
+    maresult = numpy.squeeze(x._data)
+    axes, attributes, id, grid = _extractMetadata(x)
+    return TransientVariable(
+    maresult, axes=axes, attributes=attributes, grid=grid, id=id)
+
 
 def is_masked(x):
     "Is x a 0-D masked value?"

diff --git a/Lib/avariable.py b/Lib/avariable.py
@@ -1497,6 +1497,9 @@ def getGridIndices(self):
 
     # numpy.ma overrides
 
+    def squeeze(self):
+        return(MV.squeeze(self))
+
     def __getitem__(self, key):
         if isinstance(key, tuple):
             speclist = self._process_specs(key, {})

diff --git a/Lib/tvariable.py b/Lib/tvariable.py
@@ -121,6 +121,8 @@ def __array_finalize__(self, obj):
     def __copy__(self):
         return numpy.ma.MaskedArray.copy(self)
 
+    squeeze = AbstractVariable.squeeze
+
     __mul__ = AbstractVariable.__mul__
     __rmul__ = AbstractVariable.__rmul__
     __imul__ = AbstractVariable.__imul__