allow simple lines in gnuplot.py

This thing keeps getting uglier...

1 files changed, 7 insertions, 2 deletions

diff --git a/gnuplot.py b/gnuplot.py
index d32b3c5..9262069 100755
--- a/gnuplot.py
+++ b/gnuplot.py
@@ -56,8 +56,13 @@ def gnuplot_cmd(graphs, title=None, ylabel=None, xlabel=None,
     g('set key %s' % key)
     plot = []
     for gr in graphs:
-        par = (gr[0], gr[1], gr[2], gr[3], style) if len(gr) == 4 else gr
-        plot += ["'%s' using %s:%s title '%s' with %s" % par]
+        if len(gr) == 4:
+            par = (gr[0], gr[1], gr[2], gr[3], style)
+            plot += ["'%s' using %s:%s title '%s' with %s" % par]
+        elif len(gr) == 5:
+            plot += ["'%s' using %s:%s title '%s' with %s" % gr]
+        elif len(gr) == 3:
+            plot += ["%s title '%s' with %s" % gr]
     if plot:
         g('plot ' + ', '.join(plot))
     if format != 'show' and fname: