FIX: remove colorbar from list of colorbars on axes if colorbar is removed

jklymak · jklymak · commit 64cca8e51e83 · 2021-09-03T09:35:10.000+02:00
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -992,6 +992,12 @@ def remove(self):
         If the colorbar was created with ``use_gridspec=True`` the previous
         gridspec is restored.
         """
+        if hasattr(self.ax, '_colorbar_info'):
+            parents = self.ax._colorbar_info['parents']
+            for a in parents:
+                if self.ax in a._colorbars:
+                    a._colorbars.remove(self.ax)
+
         self.ax.remove()
 
         self.mappable.callbacks.disconnect(self.mappable.colorbar_cid)
