Updated drop axis routine to squeeze axis 1 for vertex grids

modflowpy · Jun 7, 2019 · 8822069 · 8822069
1 parent bbb47de
commit 8822069
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flopy/plot/map.py b/flopy/plot/map.py
@@ -127,7 +127,7 @@ def plot_array(self, a, masked_values=None, **kwargs):
         elif self.mg.grid_type == "vertex":
             if a.ndim == 3:
                 if a.shape[0] == 1:
-                    a = np.squeeze(a, axis=0)
+                    a = np.squeeze(a, axis=1)
                     plotarray = a[self.layer, :]
                 else:
                     raise Exception("Array must be of dimension 1 or 2")
@@ -241,7 +241,7 @@ def contour_array(self, a, masked_values=None, **kwargs):
         elif self.mg.grid_type == "vertex":
             if a.ndim == 3:
                 if a.shape[0] == 1:
-                    a = np.squeeze(a, axis=0)
+                    a = np.squeeze(a, axis=1)
                     plotarray = a[self.layer, :]
                 else:
                     raise Exception("Array must be of dimension 1 or 2")