ENH: Added remove methods for legends in figure and axes objects

cimarronm · cimarronm · commit fbcaec535604 · 2014-02-09T11:43:11.000-08:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -509,6 +509,7 @@ def legend(self, *args, **kwargs):
             raise TypeError('Invalid arguments to legend.')
 
         self.legend_ = mlegend.Legend(self, handles, labels, **kwargs)
+        self.legend_._remove_method = lambda h: setattr(self, 'legend_', None)
         return self.legend_
 
     def text(self, x, y, s, fontdict=None,
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1166,6 +1166,7 @@ def legend(self, handles, labels, *args, **kwargs):
         """
         l = Legend(self, handles, labels, *args, **kwargs)
         self.legends.append(l)
+        l._remove_method = lambda h: self.legends.remove(h)
         return l
 
     @docstring.dedent_interpd
