ENH: keep track of added grispecs

jklymak · jklymak · commit 4a64f3dcbaf0 · 2018-04-11T12:27:11.000-07:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -392,6 +392,9 @@ def __init__(self,
         self._align_xlabel_grp = cbook.Grouper()
         self._align_ylabel_grp = cbook.Grouper()
 
+        # list of child gridspecs for this figure
+        self._gridspecs = []
+
     # TODO: I'd like to dynamically add the _repr_html_ method
     # to the figure in the right context, but then IPython doesn't
     # use it, for some reason.
@@ -2298,7 +2301,9 @@ def add_gridspec(self, nrows, ncols, **kwargs):
         """
 
         _ = kwargs.pop('figure', None)  # pop in case user has added this...
-        return GridSpec(nrows=nrows, ncols=ncols, figure=self, **kwargs)
+        gs = GridSpec(nrows=nrows, ncols=ncols, figure=self, **kwargs)
+        self._gridspecs.append(gs)
+        return gs
 
 
 def figaspect(arg):
