MNT: use np.argwhere

tacaswell · tacaswell · commit 0c3c61c9cc14 · 2020-02-29T20:13:52.000-05:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1641,11 +1641,11 @@ def _do_layout(gs, layout, unique_ids, nested):
             covered = np.zeros((rows, cols), dtype=bool)
             output = dict()
 
-            for j, k in zip(*np.where(layout == empty_sentinal)):
+            for j, k in np.argwhere(layout == empty_sentinal):
                 covered[j, k] = True
 
             for name in unique_ids:
-                indx = np.stack(np.where(layout == name)).T
+                indx = np.argwhere(layout == name)
                 start_row, start_col = np.min(indx, axis=0)
                 end_row, end_col = np.max(indx, axis=0) + 1
                 slc = (slice(start_row, end_row), slice(start_col, end_col))
