Make legend_handler work for stem

dstansby · dstansby · commit 6002b62f94e6 · 2017-10-25T19:08:15.000+01:00
diff --git a/lib/matplotlib/container.py b/lib/matplotlib/container.py
@@ -130,7 +130,7 @@ def __init__(self, markerline_stemlines_baseline, **kwargs):
         markerline_stemlines_baseline : tuple
             Tuple of ``(markerline, stemlines, baseline)``.
             ``markerline`` contains the `LineCollection` of the markers,
-            ``stemlines`` is a list of the `Line2D` of each line,
+            ``stemlines`` is a `LineCollection` of the main lines,
             ``baseline`` is the `Line2D` of the baseline.
         '''
         markerline, stemlines, baseline = markerline_stemlines_baseline
diff --git a/lib/matplotlib/legend_handler.py b/lib/matplotlib/legend_handler.py
@@ -542,11 +542,8 @@ def create_artists(self, legend, orig_handle,
 
         leg_stemlines = []
         for thisx, thisy in zip(xdata_marker, ydata):
-            l = Line2D([thisx, thisx], [bottom, thisy])
-            leg_stemlines.append(l)
-
-        for lm, m in zip(leg_stemlines, stemlines):
-            self.update_prop(lm, m, legend)
+            thisline = Line2D([thisx, thisx], [bottom, thisy])
+            leg_stemlines.append(thisline)
 
         leg_baseline = Line2D([np.min(xdata), np.max(xdata)],
                               [bottom, bottom])
