FIX: colorbarAxes properly in the axes stack

jklymak · jklymak · commit aaaee324fc30 · 2021-06-22T07:35:04.000-07:00
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -245,9 +245,9 @@ def __init__(self, parent, userax=True):
             True if the user passed `.Figure.colorbar` the axes manually.
         """
 
+        fig = parent.figure
         if userax:
             # copy position:
-            fig = parent.figure
             outer_ax = fig.add_axes(parent.get_position())
             # copy the locator if one exists:
             outer_ax._axes_locator = parent._axes_locator
@@ -261,11 +261,17 @@ def __init__(self, parent, userax=True):
         else:
             outer_ax = parent
 
+        # swap axes in the stack:
+        fig._localaxes.remove(outer_ax)
+        fig._axstack.remove(outer_ax)
+        fig._localaxes.add(self)
+        fig._axstack.add(self)
         inner_ax = outer_ax.inset_axes([0, 0, 1, 1])
         self.__dict__.update(inner_ax.__dict__)
 
         self.outer_ax = outer_ax
         self.inner_ax = inner_ax
+
         self.outer_ax.xaxis.set_visible(False)
         self.outer_ax.yaxis.set_visible(False)
         self.outer_ax.set_facecolor('none')
@@ -285,6 +291,9 @@ def _set_inner_bounds(self, bounds):
         self.inner_ax._axes_locator = _TransformedBoundsLocator(
             bounds, self.outer_ax.transAxes)
 
+    def draw(self, renderer):
+        self.outer_ax.draw(renderer)
+
 
 class _ColorbarSpine(mspines.Spine):
     def __init__(self, axes):
