Add pie container to Axes.containers list

rcomer · rcomer · commit 82b1c1ba6a1d · 2025-11-18T20:09:17.000Z
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -3710,6 +3710,7 @@ def get_next_color():
             theta1 = theta2
 
         pc = PieContainer(slices, x, normalize)
+        self.add_container(pc)
 
         if labeldistance is None:
             # Insert an empty list of texts for backwards compatibility of the
diff --git a/lib/matplotlib/container.py b/lib/matplotlib/container.py
@@ -181,6 +181,7 @@ def __init__(self, wedges, values, normalize):
         self._texts = []
         self._values = values
         self._normalize = normalize
+        self._remove_method = None
 
     @property
     def texts(self):
@@ -214,11 +215,19 @@ def remove(self):
             for artist in cbook.flatten(artist_list):
                 artist.remove()
 
+        if self._remove_method is not None:
+            self._remove_method(self)
+
     def __getitem__(self, key):
         # needed to support unpacking into a tuple for backward compatibility of the
         # Axes.pie return value
         return (self.wedges, *self._texts)[key]
 
+    def get_label(self):
+        # needed for adding to the Axes.container list but we have no legend handler
+        # so always return a string that legend will ignore
+        return '_pie'
+
 
 class StemContainer(Container):
     """
diff --git a/lib/matplotlib/tests/test_container.py b/lib/matplotlib/tests/test_container.py
@@ -61,10 +61,12 @@ def test_piecontainer_remove():
     ax.pie_label(pie, ['baz', 'qux'])
     assert len(ax.patches) == 2
     assert len(ax.texts) == 6
+    assert pie in ax.containers
 
     pie.remove()
     assert not ax.patches
     assert not ax.texts
+    assert not ax.containers
 
 
 def test_piecontainer_unpack_backcompat():
