Better file processing in plot_pm and plotting errorbars

1 files changed, 9 insertions, 2 deletions

diff --git a/gnuplot.py b/gnuplot.py
index d32b3c5..0afb9b3 100755
--- a/gnuplot.py
+++ b/gnuplot.py
@@ -56,8 +56,15 @@ 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) == 6:
+            par = (gr[0], gr[1], gr[2], gr[3], gr[4], style, gr[5])
+            plot += ["'%s' using %s:%s:%s:%s with %s title '%s'" % par]
+        else:
+            par = gr
+            plot += ["'%s' using %s:%s title '%s' with %s" % par]
     if plot:
         g('plot ' + ', '.join(plot))
     if format != 'show' and fname: