Merge pull request #24627 from tacaswell/mnt/cla_clear_chlildren

anntzer · web-flow · commit b2e843405de3 · 2022-12-07T23:23:54.000+01:00
MNT: when clearing an Axes via clear/cla fully detach children
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -1300,7 +1300,9 @@ def __clear(self):
         self._get_patches_for_fill = _process_plot_var_args(self, 'fill')
 
         self._gridOn = mpl.rcParams['axes.grid']
-        self._children = []
+        old_children, self._children = self._children, []
+        for chld in old_children:
+            chld.axes = chld.figure = None
         self._mouseover_set = _OrderedSet()
         self.child_axes = []
         self._current_image = None  # strictly for pyplot via _sci, _gci
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -8370,6 +8370,19 @@ def test_extent_units():
         im.set_extent([2, 12, date_first, date_last], clip=False)
 
 
+def test_cla_clears_children_axes_and_fig():
+    fig, ax = plt.subplots()
+    lines = ax.plot([], [], [], [])
+    img = ax.imshow([[1]])
+    for art in lines + [img]:
+        assert art.axes is ax
+        assert art.figure is fig
+    ax.clear()
+    for art in lines + [img]:
+        assert art.axes is None
+        assert art.figure is None
+
+
 def test_scatter_color_repr_error():
 
     def get_next_color():
