Merge pull request #20414 from jklymak/enh-colorbar_info

QuLogic · web-flow · commit 6014058cafc9 · 2021-06-16T18:04:41.000-04:00
ENH: add colorbar info to gridspec cbar
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -274,7 +274,9 @@ def __init__(self, parent, userax=True):
         self.outer_ax.set_yticks = self.inner_ax.set_yticks
         for attr in ["get_position", "set_position", "set_aspect"]:
             setattr(self, attr, getattr(self.outer_ax, attr))
+        self._colorbar_info = None  # used for mpl-created axes
         if userax:
+            self._colorbar_info = 'user'
             # point the parent's methods all at this axes...
             parent.__dict__ = self.__dict__
 
@@ -1487,6 +1489,15 @@ def make_axes_gridspec(parent, *, location=None, orientation=None,
     fig = parent.get_figure()
     cax = fig.add_subplot(ss_cb, label="<colorbar>")
     cax.set_aspect(aspect, anchor=loc_settings["anchor"], adjustable='box')
+    cax._colorbar_info = dict(
+        location=location,
+        parents=[parent],
+        shrink=shrink,
+        anchor=anchor,
+        panchor=panchor,
+        fraction=fraction,
+        aspect=aspect,
+        pad=pad)
     return cax, kw
 
 
