From 1f713d2a7ba4ded6369c45b819db3b78a9735c0f Mon Sep 17 00:00:00 2001
From: kojix2 <2xijok@gmail.com>
Date: Fri, 14 Aug 2020 10:36:27 +0900
Subject: [PATCH] Refactor line plot

---
 lib/uplot/command.rb | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/lib/uplot/command.rb b/lib/uplot/command.rb
index 7d70273..4282feb 100644
--- a/lib/uplot/command.rb
+++ b/lib/uplot/command.rb
@@ -173,17 +173,15 @@ module Uplot
     def line(data, headers)
       if data.size == 1
         @params[:ylabel] ||= headers[0] if headers
-        y = data[0]
-        x = (1..y.size).to_a
+        y = data[0].map(&:to_f)
+        UnicodePlot.lineplot(y, **@params.compact)
       else
         @params[:xlabel] ||= headers[0] if headers
         @params[:ylabel] ||= headers[1] if headers
-        x = data[0]
-        y = data[1]
+        x = data[0].map(&:to_f)
+        y = data[1].map(&:to_f)
+        UnicodePlot.lineplot(x, y, **@params.compact)
       end
-      x = x.map(&:to_f)
-      y = y.map(&:to_f)
-      UnicodePlot.lineplot(x, y, **@params.compact)
     end
 
     def get_method2(method1)