Added function to check if line is in legend, then use *that* to plot, or not plot, a legend entry for a line

Author: spott

matplotlib2tikz/line2d.py

@@ -3,6 +3,17 @@
 from . import color as mycol
 from . import path as mypath
 
+def get_legend_label_(line):
+    '''Check if line is in legend
+    '''
+
+    label = line.get_label()
+    try:
+        ax = line.axes
+        leg = ax.get_legend()
+        return label in [l.get_label() for l in leg.get_lines()]
+    except:
+        return None
 
 def draw_line2d(data, obj):
     '''Returns the PGFPlots code for an Line2D environment.
@@ -92,7 +103,7 @@ def draw_line2d(data, obj):
 
     # Check if a line is not in a legend and forget it if so,
     # fixes bug #167:
-    if obj.get_label().startswith("_line"):
+    if not get_legend_label_(obj):
         addplot_options.append("forget plot")
 
     # process options