ENH: make it work with axes objects not strings

tacaswell · tacaswell · commit 6233726561e1 · 2020-02-28T04:47:00.000-05:00
Update the objects in place
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1622,7 +1622,7 @@ def _process_layout(layout):
             nested = {}
             for j, row in enumerate(layout):
                 for k, v in enumerate(row):
-                    if isinstance(v, str):
+                    if isinstance(v, (str, Axes)):
                         unique_ids.add(v)
                     else:
                         nested[(j, k)] = v
@@ -1641,7 +1641,13 @@ def _do_layout(gs, layout, unique_ids, nested):
                 if not np.all(covered[slc] == False):
                     raise ValueError
                 covered[slc] = True
-                output[name] = self.add_subplot(gs[slc], **subplot_kw)
+                if isinstance(name, str):
+                    output[name] = self.add_subplot(gs[slc], **subplot_kw)
+                else:
+                    ax = name
+                    ax.set_subplotspec(gs[slc])
+                    ax.update_params()
+                    ax.set_position(ax.figbox)
 
             for (j, k), layout in nested.items():
                 layout = np.asarray(layout)
