Detach legend when clearing an Axes via clear/cla

daniilS · daniilS · commit 0bcfe16f1783 · 2023-07-25T01:34:41.000+01:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -1298,6 +1298,9 @@ def __clear(self):
         self.child_axes = []
         self._current_image = None  # strictly for pyplot via _sci, _gci
         self._projection_init = None  # strictly for pyplot.subplot
+
+        if hasattr(self, "legend_") and self.legend_ is not None:
+            self.legend_.axes = self.legend_.figure = None
         self.legend_ = None
         self.containers = []
 
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -8644,13 +8644,19 @@ def test_extent_units():
 
 def test_cla_clears_children_axes_and_fig():
     fig, ax = plt.subplots()
-    lines = ax.plot([], [], [], [])
+
+    lines = ax.plot([], [], [], [], label="line")
     img = ax.imshow([[1]])
-    for art in lines + [img]:
+    legend = ax.legend()
+
+    children = [*lines, img, legend]
+
+    for art in children:
         assert art.axes is ax
         assert art.figure is fig
+
     ax.clear()
-    for art in lines + [img]:
+    for art in children:
         assert art.axes is None
         assert art.figure is None
 
