Add QuadContourSet.remove.

anntzer · anntzer · commit 17e5a80d394a · 2022-10-11T22:09:01.000+02:00
diff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py
@@ -585,6 +585,10 @@ def labels(self, inline, inline_spacing):
             if inline:
                 paths[:] = additions
 
+    def remove(self):
+        for text in self.labelTexts:
+            text.remove()
+
 
 def _is_closed_polygon(X):
     """
@@ -1389,6 +1393,11 @@ def find_nearest_contour(self, x, y, indices=None, pixel=True):
 
         return (conmin, segmin, imin, xmin, ymin, d2min)
 
+    def remove(self):
+        super().remove()
+        for coll in self.collections:
+            coll.remove()
+
 
 @_docstring.dedent_interpd
 class QuadContourSet(ContourSet):
diff --git a/lib/matplotlib/tests/test_contour.py b/lib/matplotlib/tests/test_contour.py
@@ -682,3 +682,13 @@ def test_negative_linestyles(style):
     ax4.clabel(CS4, fontsize=9, inline=True)
     ax4.set_title(f'Single color - negative contours {style}')
     assert CS4.negative_linestyles == style
+
+
+def test_contour_remove():
+    ax = plt.figure().add_subplot()
+    orig_children = ax.get_children()
+    cs = ax.contour(np.arange(16).reshape((4, 4)))
+    cs.clabel()
+    assert ax.get_children() != orig_children
+    cs.remove()
+    assert ax.get_children() == orig_children
