mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-09-18 18:08:07 +08:00
Compare commits
75 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
e0914beec8 | ||
![]() |
e146bc66f3 | ||
![]() |
bc0204af53 | ||
![]() |
286e90ec23 | ||
![]() |
f3cd03196d | ||
![]() |
c4d31108bb | ||
![]() |
9d58b1aaf9 | ||
![]() |
a09a703c33 | ||
![]() |
4a14e4716a | ||
![]() |
3c9f2fc9fc | ||
![]() |
c0f5cbc7d5 | ||
![]() |
57efdfcc2c | ||
![]() |
d17143f7fa | ||
![]() |
c426378bda | ||
![]() |
02857a46a3 | ||
![]() |
ff7b0680aa | ||
![]() |
e887dc3f5a | ||
![]() |
1d1ae9e1b0 | ||
![]() |
715236b5b4 | ||
![]() |
22460f3689 | ||
![]() |
1548675967 | ||
![]() |
45fc89605c | ||
![]() |
ee3608a106 | ||
![]() |
ff1176797f | ||
![]() |
6db2372c29 | ||
![]() |
59f0fdddca | ||
![]() |
3c51e0c902 | ||
![]() |
ee74b32815 | ||
![]() |
0fa1e37da7 | ||
![]() |
3c6a4b46b1 | ||
![]() |
1f713d2a7b | ||
![]() |
749ac8a091 | ||
![]() |
0906becf27 | ||
![]() |
cd4f7281ef | ||
![]() |
b24d91642b | ||
![]() |
47ba4ededa | ||
![]() |
1ac39003bc | ||
![]() |
2fddc98118 | ||
![]() |
edb377170d | ||
![]() |
0d7bac71d8 | ||
![]() |
7e42caa506 | ||
![]() |
1dec36641b | ||
![]() |
eadafa6307 | ||
![]() |
49f8010235 | ||
![]() |
78559b989e | ||
![]() |
d069f8af23 | ||
![]() |
0c1fcc517b | ||
![]() |
6a00314fc4 | ||
![]() |
76f88eb55a | ||
![]() |
1dce48cd2c | ||
![]() |
41a540c876 | ||
![]() |
868ab0a197 | ||
![]() |
71afa3cda7 | ||
![]() |
305489d591 | ||
![]() |
9d6337df1c | ||
![]() |
5557c4c1d0 | ||
![]() |
76d3f46549 | ||
![]() |
7c0ab3ebdb | ||
![]() |
1fb369e26b | ||
![]() |
d72c084602 | ||
![]() |
50de2bac48 | ||
![]() |
eabc36b581 | ||
![]() |
5f979d2e85 | ||
![]() |
8536abc061 | ||
![]() |
b4856e45d0 | ||
![]() |
87a8af4957 | ||
![]() |
09271ab6c3 | ||
![]() |
0110b70936 | ||
![]() |
9859cd0213 | ||
![]() |
cb43796e3f | ||
![]() |
958ce45635 | ||
![]() |
203e48e0ed | ||
![]() |
155a0821d3 | ||
![]() |
4a49711d57 | ||
![]() |
e69d1caed7 |
2
.github/FUNDING.yml
vendored
Normal file
2
.github/FUNDING.yml
vendored
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
ko_fi: kojix2
|
||||||
|
patreon: kojix2
|
1
.gitignore
vendored
1
.gitignore
vendored
@@ -6,4 +6,5 @@
|
|||||||
/pkg/
|
/pkg/
|
||||||
/spec/reports/
|
/spec/reports/
|
||||||
/tmp/
|
/tmp/
|
||||||
|
/vendor/
|
||||||
*.lock
|
*.lock
|
||||||
|
3
Gemfile
3
Gemfile
@@ -2,6 +2,3 @@ source 'https://rubygems.org'
|
|||||||
|
|
||||||
# Specify your gem's dependencies in uplot.gemspec
|
# Specify your gem's dependencies in uplot.gemspec
|
||||||
gemspec
|
gemspec
|
||||||
|
|
||||||
gem 'minitest', '~> 5.0'
|
|
||||||
gem 'rake', '~> 12.0'
|
|
||||||
|
20
README.md
20
README.md
@@ -1,13 +1,15 @@
|
|||||||
# Uplot
|
# uplot
|
||||||
|
|
||||||
[](https://travis-ci.com/kojix2/uplot)
|
[](https://travis-ci.com/kojix2/uplot)
|
||||||
[](https://badge.fury.io/rb/u-plot)
|
[](https://badge.fury.io/rb/u-plot)
|
||||||
|
[](https://rubydoc.info/gems/u-plot)
|
||||||
|
[](LICENSE.txt)
|
||||||
|
|
||||||
Create ASCII charts on the terminal with data from standard streams in the pipeline.
|
Create ASCII charts on the terminal with data from standard streams in the pipeline.
|
||||||
|
|
||||||
:bar_chart: Powered by [UnicodePlot](https://github.com/kojix2/uplot)
|
:bar_chart: Powered by [UnicodePlot](https://github.com/red-data-tools/unicode_plot.rb)
|
||||||
|
|
||||||
:construction: Under development
|
:construction: Under development! :construction:
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
@@ -17,7 +19,7 @@ gem install u-plot
|
|||||||
|
|
||||||
## Usage
|
## Usage
|
||||||
|
|
||||||
### histogram
|
**histogram**
|
||||||
|
|
||||||
```sh
|
```sh
|
||||||
ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
|
ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
|
||||||
@@ -46,10 +48,20 @@ ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
|
|||||||
Frequency
|
Frequency
|
||||||
```
|
```
|
||||||
|
|
||||||
|
**scatter**
|
||||||
|
|
||||||
|
```sh
|
||||||
|
wget https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv -qO - \
|
||||||
|
| cut -f1-4 -d, \
|
||||||
|
| uplot scatter -H -d, -t IRIS -m 10
|
||||||
|
```
|
||||||
|
|
||||||
## Development
|
## Development
|
||||||
|
|
||||||
## Contributing
|
## Contributing
|
||||||
|
|
||||||
|
Bug reports and pull requests are welcome on GitHub at [https://github.com/kojix2/uplot](https://github.com/kojix2/uplot).
|
||||||
|
|
||||||
## License
|
## License
|
||||||
|
|
||||||
The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
|
The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
|
||||||
|
@@ -2,4 +2,4 @@
|
|||||||
|
|
||||||
require 'uplot'
|
require 'uplot'
|
||||||
|
|
||||||
Uplot::Command.new(ARGV).run
|
Uplot::Command.new.run
|
||||||
|
46
lib/uplot.rb
46
lib/uplot.rb
@@ -1,46 +1,8 @@
|
|||||||
require 'uplot/version'
|
|
||||||
require 'unicode_plot'
|
require 'unicode_plot'
|
||||||
require 'optparse'
|
require 'uplot/version'
|
||||||
|
require 'uplot/preprocessing'
|
||||||
|
require 'uplot/plot'
|
||||||
|
require 'uplot/command'
|
||||||
|
|
||||||
module Uplot
|
module Uplot
|
||||||
class Command
|
|
||||||
def initialize(argv)
|
|
||||||
@params = {}
|
|
||||||
@ptype = nil
|
|
||||||
parse_options(argv)
|
|
||||||
end
|
|
||||||
|
|
||||||
def parse_options(argv)
|
|
||||||
parser = OptionParser.new
|
|
||||||
parser.order!(argv)
|
|
||||||
@ptype = argv.shift
|
|
||||||
|
|
||||||
subparsers = Hash.new do |_h, k|
|
|
||||||
warn "no such subcommand: #{k}"
|
|
||||||
exit 1
|
|
||||||
end
|
|
||||||
|
|
||||||
subparsers['hist'] = OptionParser.new.tap do |sub|
|
|
||||||
sub.on('--nbins VAL') { |v| @params[:nbins] = v.to_i }
|
|
||||||
sub.on('-p') { |v| @params[:p] = v }
|
|
||||||
end
|
|
||||||
|
|
||||||
subparsers[@ptype].parse!(argv) unless argv.empty?
|
|
||||||
end
|
|
||||||
|
|
||||||
def run
|
|
||||||
input_lines = readlines.map(&:chomp)
|
|
||||||
case @ptype
|
|
||||||
when 'hist', 'histogram'
|
|
||||||
histogram(input_lines).render
|
|
||||||
end
|
|
||||||
|
|
||||||
puts input_lines if @params[:p]
|
|
||||||
end
|
|
||||||
|
|
||||||
def histogram(input_lines)
|
|
||||||
series = input_lines.map(&:to_f)
|
|
||||||
UnicodePlot.histogram(series, nbins: @params[:nbins])
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
end
|
||||||
|
276
lib/uplot/command.rb
Normal file
276
lib/uplot/command.rb
Normal file
@@ -0,0 +1,276 @@
|
|||||||
|
require 'optparse'
|
||||||
|
require_relative 'preprocessing'
|
||||||
|
|
||||||
|
module Uplot
|
||||||
|
Data = Struct.new(:headers, :series)
|
||||||
|
|
||||||
|
class Command
|
||||||
|
Params = Struct.new(
|
||||||
|
# Sort me!
|
||||||
|
:title,
|
||||||
|
:width,
|
||||||
|
:height,
|
||||||
|
:border,
|
||||||
|
:margin,
|
||||||
|
:padding,
|
||||||
|
:color,
|
||||||
|
:xlabel,
|
||||||
|
:ylabel,
|
||||||
|
:labels,
|
||||||
|
:symbol,
|
||||||
|
:xscale,
|
||||||
|
:nbins,
|
||||||
|
:closed,
|
||||||
|
:canvas,
|
||||||
|
:xlim,
|
||||||
|
:ylim,
|
||||||
|
:grid,
|
||||||
|
:name
|
||||||
|
) do
|
||||||
|
def to_hc
|
||||||
|
to_h.compact
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
attr_accessor :params, :plot_type
|
||||||
|
attr_reader :raw_inputs, :data, :fmt
|
||||||
|
|
||||||
|
def initialize
|
||||||
|
@params = Params.new
|
||||||
|
|
||||||
|
@plot_type = nil
|
||||||
|
@headers = nil
|
||||||
|
@delimiter = "\t"
|
||||||
|
@transpose = false
|
||||||
|
@output = false
|
||||||
|
@count = false
|
||||||
|
@fmt = 'xyy'
|
||||||
|
|
||||||
|
@raw_inputs = []
|
||||||
|
@debug = false
|
||||||
|
end
|
||||||
|
|
||||||
|
def create_parser
|
||||||
|
OptionParser.new do |opt|
|
||||||
|
opt.program_name = 'uplot'
|
||||||
|
opt.version = Uplot::VERSION
|
||||||
|
opt.on('-o', '--output', TrueClass) do |v|
|
||||||
|
@output = v
|
||||||
|
end
|
||||||
|
.on('-d', '--delimiter VAL', String) do |v|
|
||||||
|
@delimiter = v
|
||||||
|
end
|
||||||
|
.on('-H', '--headers', TrueClass) do |v|
|
||||||
|
@headers = v
|
||||||
|
end
|
||||||
|
.on('-T', '--transpose', TrueClass) do |v|
|
||||||
|
@transpose = v
|
||||||
|
end
|
||||||
|
.on('-t', '--title VAL', String) do |v|
|
||||||
|
params.title = v
|
||||||
|
end
|
||||||
|
.on('-w', '--width VAL', Numeric) do |v|
|
||||||
|
params.width = v
|
||||||
|
end
|
||||||
|
.on('-h', '--height VAL', Numeric) do |v|
|
||||||
|
params.height = v
|
||||||
|
end
|
||||||
|
.on('-b', '--border VAL', Numeric) do |v|
|
||||||
|
params.border = v
|
||||||
|
end
|
||||||
|
.on('-m', '--margin VAL', Numeric) do |v|
|
||||||
|
params.margin = v
|
||||||
|
end
|
||||||
|
.on('-p', '--padding VAL', Numeric) do |v|
|
||||||
|
params.padding = v
|
||||||
|
end
|
||||||
|
.on('-c', '--color VAL', String) do |v|
|
||||||
|
params.color = v.to_sym
|
||||||
|
end
|
||||||
|
.on('-x', '--xlabel VAL', String) do |v|
|
||||||
|
params.xlabel = v
|
||||||
|
end
|
||||||
|
.on('-y', '--ylabel VAL', String) do |v|
|
||||||
|
params.ylabel = v
|
||||||
|
end
|
||||||
|
.on('-l', '--labels', TrueClass) do |v|
|
||||||
|
params.labels = v
|
||||||
|
end
|
||||||
|
.on('--fmt VAL', String) do |v|
|
||||||
|
@fmt = v
|
||||||
|
end
|
||||||
|
.on('--debug', TrueClass) do |v|
|
||||||
|
@debug = v
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def parse_options(argv = ARGV)
|
||||||
|
main_parser = create_parser
|
||||||
|
parsers = Hash.new { |h, k| h[k] = create_parser }
|
||||||
|
|
||||||
|
parsers[:barplot] = \
|
||||||
|
parsers[:bar]
|
||||||
|
.on('--symbol VAL', String) do |v|
|
||||||
|
params.symbol = v
|
||||||
|
end
|
||||||
|
.on('--xscale VAL', String) do |v|
|
||||||
|
params.xscale = v
|
||||||
|
end
|
||||||
|
.on('--count', TrueClass) do |v|
|
||||||
|
@count = v
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:count] = \
|
||||||
|
parsers[:c] # barplot -c
|
||||||
|
.on('--symbol VAL', String) do |v|
|
||||||
|
params.symbol = v
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:histogram] = \
|
||||||
|
parsers[:hist]
|
||||||
|
.on('-n', '--nbins VAL', Numeric) do |v|
|
||||||
|
params.nbins = v
|
||||||
|
end
|
||||||
|
.on('--closed VAL', String) do |v|
|
||||||
|
params.closed = v
|
||||||
|
end
|
||||||
|
.on('--symbol VAL', String) do |v|
|
||||||
|
params.symbol = v
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:lineplot] = \
|
||||||
|
parsers[:line]
|
||||||
|
.on('--canvas VAL', String) do |v|
|
||||||
|
params.canvas = v
|
||||||
|
end
|
||||||
|
.on('--xlim VAL', String) do |v|
|
||||||
|
params.xlim = get_lim(v)
|
||||||
|
end
|
||||||
|
.on('--ylim VAL', String) do |v|
|
||||||
|
params.ylim = get_lim(v)
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:lineplots] = \
|
||||||
|
parsers[:lines]
|
||||||
|
.on('--canvas VAL', String) do |v|
|
||||||
|
params.canvas = v
|
||||||
|
end
|
||||||
|
.on('--xlim VAL', String) do |v|
|
||||||
|
params.xlim = get_lim(v)
|
||||||
|
end
|
||||||
|
.on('--ylim VAL', String) do |v|
|
||||||
|
params.ylim = get_lim(v)
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:scatter] = \
|
||||||
|
parsers[:s]
|
||||||
|
.on('--canvas VAL', String) do |v|
|
||||||
|
params.canvas = v
|
||||||
|
end
|
||||||
|
.on('--xlim VAL', String) do |v|
|
||||||
|
params.xlim = get_lim(v)
|
||||||
|
end
|
||||||
|
.on('--ylim VAL', String) do |v|
|
||||||
|
params.ylim = get_lim(v)
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:density] = \
|
||||||
|
parsers[:d]
|
||||||
|
.on('--grid', TrueClass) do |v|
|
||||||
|
params.grid = v
|
||||||
|
end
|
||||||
|
.on('--xlim VAL', String) do |v|
|
||||||
|
params.xlim = get_lim(v)
|
||||||
|
end
|
||||||
|
.on('--ylim VAL', String) do |v|
|
||||||
|
params.ylim = get_lim(v)
|
||||||
|
end
|
||||||
|
|
||||||
|
parsers[:boxplot] = \
|
||||||
|
parsers[:box]
|
||||||
|
.on('--xlim VAL', String) do |v|
|
||||||
|
params.xlim = get_lim(v)
|
||||||
|
end
|
||||||
|
|
||||||
|
# Preventing the generation of new sub-commands
|
||||||
|
parsers.default = nil
|
||||||
|
|
||||||
|
# Usage and help messages
|
||||||
|
main_parser.banner = \
|
||||||
|
<<~MSG
|
||||||
|
Program: uplot (Tools for plotting on the terminal)
|
||||||
|
Version: #{Uplot::VERSION} (using unicode_plot #{UnicodePlot::VERSION})
|
||||||
|
|
||||||
|
Usage: uplot <command> [options]
|
||||||
|
|
||||||
|
Command: #{parsers.keys.join(' ')}
|
||||||
|
|
||||||
|
Options:
|
||||||
|
MSG
|
||||||
|
|
||||||
|
begin
|
||||||
|
main_parser.order!(argv)
|
||||||
|
rescue OptionParser::ParseError => e
|
||||||
|
warn "uplot: #{e.message}"
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
|
||||||
|
@plot_type = argv.shift&.to_sym
|
||||||
|
|
||||||
|
unless parsers.has_key?(plot_type)
|
||||||
|
if plot_type.nil?
|
||||||
|
warn main_parser.help
|
||||||
|
else
|
||||||
|
warn "uplot: unrecognized command '#{plot_type}'"
|
||||||
|
end
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
parser = parsers[plot_type]
|
||||||
|
|
||||||
|
begin
|
||||||
|
parser.parse!(argv) unless argv.empty?
|
||||||
|
rescue OptionParser::ParseError => e
|
||||||
|
warn "uplot: #{e.message}"
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def get_lim(str)
|
||||||
|
str.split(/-|:|\.\./)[0..1].map(&:to_f)
|
||||||
|
end
|
||||||
|
|
||||||
|
def run
|
||||||
|
parse_options
|
||||||
|
# Sometimes the input file does not end with a newline code.
|
||||||
|
while input = Kernel.gets(nil)
|
||||||
|
input.freeze
|
||||||
|
@raw_inputs << input
|
||||||
|
@data = Preprocessing.input(input, @delimiter, @headers, @transpose)
|
||||||
|
pp @data if @debug
|
||||||
|
case plot_type
|
||||||
|
when :bar, :barplot
|
||||||
|
Plot.barplot(data, params, @count)
|
||||||
|
when :count, :c
|
||||||
|
Plot.barplot(data, params, count = true)
|
||||||
|
when :hist, :histogram
|
||||||
|
Plot.histogram(data, params)
|
||||||
|
when :line, :lineplot
|
||||||
|
Plot.line(data, params)
|
||||||
|
when :lines, :lineplots
|
||||||
|
Plot.lines(data, params, fmt)
|
||||||
|
when :scatter, :s
|
||||||
|
Plot.scatter(data, params, fmt)
|
||||||
|
when :density, :d
|
||||||
|
Plot.density(data, params, fmt)
|
||||||
|
when :box, :boxplot
|
||||||
|
Plot.boxplot(data, params)
|
||||||
|
else
|
||||||
|
raise "unrecognized plot_type: #{plot_type}"
|
||||||
|
end.render($stderr)
|
||||||
|
|
||||||
|
print input if @output
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
145
lib/uplot/plot.rb
Normal file
145
lib/uplot/plot.rb
Normal file
@@ -0,0 +1,145 @@
|
|||||||
|
require 'unicode_plot'
|
||||||
|
|
||||||
|
module Uplot
|
||||||
|
# plotting functions.
|
||||||
|
module Plot
|
||||||
|
module_function
|
||||||
|
|
||||||
|
def barplot(data, params, count = false)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
if count
|
||||||
|
series = Preprocessing.count(series[0])
|
||||||
|
params.title = headers[0] if headers
|
||||||
|
end
|
||||||
|
params.title ||= headers[1] if headers
|
||||||
|
labels = series[0]
|
||||||
|
values = series[1].map(&:to_f)
|
||||||
|
UnicodePlot.barplot(labels, values, **params.to_hc)
|
||||||
|
end
|
||||||
|
|
||||||
|
def histogram(data, params)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
params.title ||= data.headers[0] if headers
|
||||||
|
values = series[0].map(&:to_f)
|
||||||
|
UnicodePlot.histogram(values, **params.to_hc)
|
||||||
|
end
|
||||||
|
|
||||||
|
def line(data, params)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
if series.size == 1
|
||||||
|
# If there is only one series, it is assumed to be sequential data.
|
||||||
|
params.ylabel ||= headers[0] if headers
|
||||||
|
y = series[0].map(&:to_f)
|
||||||
|
UnicodePlot.lineplot(y, **params.to_hc)
|
||||||
|
else
|
||||||
|
# If there are 2 or more series,
|
||||||
|
# assume that the first 2 series are the x and y series respectively.
|
||||||
|
if headers
|
||||||
|
params.xlabel ||= headers[0]
|
||||||
|
params.ylabel ||= headers[1]
|
||||||
|
end
|
||||||
|
x = series[0].map(&:to_f)
|
||||||
|
y = series[1].map(&:to_f)
|
||||||
|
UnicodePlot.lineplot(x, y, **params.to_hc)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def get_method2(method1)
|
||||||
|
(method1.to_s + '!').to_sym
|
||||||
|
end
|
||||||
|
|
||||||
|
def xyy_plot(data, method1, params)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
method2 = get_method2(method1)
|
||||||
|
series.map! { |s| s.map(&:to_f) }
|
||||||
|
if headers
|
||||||
|
params.name ||= headers[1]
|
||||||
|
params.xlabel ||= headers[0]
|
||||||
|
end
|
||||||
|
params.ylim ||= series[1..-1].flatten.minmax # why need?
|
||||||
|
plot = UnicodePlot.public_send(method1, series[0], series[1], **params.to_hc)
|
||||||
|
2.upto(series.size - 1) do |i|
|
||||||
|
UnicodePlot.public_send(method2, plot, series[0], series[i], name: headers&.[](i))
|
||||||
|
end
|
||||||
|
plot
|
||||||
|
end
|
||||||
|
|
||||||
|
def xyxy_plot(data, method1, params)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
method2 = get_method2(method1)
|
||||||
|
series.map! { |s| s.map(&:to_f) }
|
||||||
|
series = series.each_slice(2).to_a
|
||||||
|
params.name ||= headers[0] if headers
|
||||||
|
params.xlim = series.map(&:first).flatten.minmax # why need?
|
||||||
|
params.ylim = series.map(&:last).flatten.minmax # why need?
|
||||||
|
x1, y1 = series.shift
|
||||||
|
plot = UnicodePlot.public_send(method1, x1, y1, **params.to_hc)
|
||||||
|
series.each_with_index do |(xi, yi), i|
|
||||||
|
UnicodePlot.public_send(method2, plot, xi, yi, name: headers&.[]((i + 1) * 2))
|
||||||
|
end
|
||||||
|
plot
|
||||||
|
end
|
||||||
|
|
||||||
|
def lines(data, params, fmt = 'xyy')
|
||||||
|
check_series_size(data, fmt)
|
||||||
|
case fmt
|
||||||
|
when 'xyy'
|
||||||
|
xyy_plot(data, :lineplot, params)
|
||||||
|
when 'xyxy'
|
||||||
|
xyxy_plot(data, :lineplot, params)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def scatter(data, params, fmt = 'xyy')
|
||||||
|
check_series_size(data, fmt)
|
||||||
|
case fmt
|
||||||
|
when 'xyy'
|
||||||
|
xyy_plot(data, :scatterplot, params)
|
||||||
|
when 'xyxy'
|
||||||
|
xyxy_plot(data, :scatterplot, params)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def density(data, params, fmt = 'xyy')
|
||||||
|
check_series_size(data, fmt)
|
||||||
|
case fmt
|
||||||
|
when 'xyy'
|
||||||
|
xyy_plot(data, :densityplot, params)
|
||||||
|
when 'xyxy'
|
||||||
|
xyxy_plot(data, :densityplot, params)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def boxplot(data, params)
|
||||||
|
headers = data.headers
|
||||||
|
series = data.series
|
||||||
|
headers ||= (1..series.size).map(&:to_s)
|
||||||
|
series.map! { |s| s.map(&:to_f) }
|
||||||
|
UnicodePlot.boxplot(headers, series, **params.to_hc)
|
||||||
|
end
|
||||||
|
|
||||||
|
def check_series_size(data, fmt)
|
||||||
|
series = data.series
|
||||||
|
if series.size == 1
|
||||||
|
warn 'uplot: There is only one series of input data. Please check the delimiter.'
|
||||||
|
warn ''
|
||||||
|
warn " Headers: \e[35m#{data.headers.inspect}\e[0m"
|
||||||
|
warn " The first item is: \e[35m\"#{series[0][0]}\"\e[0m"
|
||||||
|
warn " The last item is : \e[35m\"#{series[0][-1]}\"\e[0m"
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
if fmt == 'xyxy' && series.size.odd?
|
||||||
|
warn 'uplot: In the xyxy format, the number of series must be even.'
|
||||||
|
warn ''
|
||||||
|
warn " Number of series: \e[35m#{series.size}\e[0m"
|
||||||
|
warn " Headers: \e[35m#{data.headers.inspect}\e[0m"
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
67
lib/uplot/preprocessing.rb
Normal file
67
lib/uplot/preprocessing.rb
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
require 'csv'
|
||||||
|
|
||||||
|
module Uplot
|
||||||
|
module Preprocessing
|
||||||
|
module_function
|
||||||
|
|
||||||
|
def input(input, delimiter, headers, transpose)
|
||||||
|
arr = read_csv(input, delimiter)
|
||||||
|
headers = get_headers(arr, headers, transpose)
|
||||||
|
series = get_series(arr, headers, transpose)
|
||||||
|
Data.new(headers, series)
|
||||||
|
end
|
||||||
|
|
||||||
|
def read_csv(input, delimiter)
|
||||||
|
CSV.parse(input, col_sep: delimiter)
|
||||||
|
.delete_if do |i|
|
||||||
|
i == [] or i.all? nil
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Transpose different sized ruby arrays
|
||||||
|
# https://stackoverflow.com/q/26016632
|
||||||
|
def transpose2(arr)
|
||||||
|
Array.new(arr.map(&:length).max) { |i| arr.map { |e| e[i] } }
|
||||||
|
end
|
||||||
|
|
||||||
|
def get_headers(arr, headers, transpose)
|
||||||
|
if headers
|
||||||
|
if transpose
|
||||||
|
arr.map(&:first)
|
||||||
|
else
|
||||||
|
arr[0]
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def get_series(arr, headers, transpose)
|
||||||
|
if transpose
|
||||||
|
if headers
|
||||||
|
arr.map { |row| row[1..-1] }
|
||||||
|
else
|
||||||
|
arr
|
||||||
|
end
|
||||||
|
else
|
||||||
|
if headers
|
||||||
|
transpose2(arr[1..-1])
|
||||||
|
else
|
||||||
|
transpose2(arr)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def count(arr)
|
||||||
|
# tally was added in Ruby 2.7
|
||||||
|
if Enumerable.method_defined? :tally
|
||||||
|
arr.tally
|
||||||
|
else
|
||||||
|
# https://github.com/marcandre/backports
|
||||||
|
arr.each_with_object(Hash.new(0)) { |item, res| res[item] += 1 }
|
||||||
|
.tap { |h| h.default = nil }
|
||||||
|
end
|
||||||
|
.sort { |a, b| a[1] <=> b[1] }
|
||||||
|
.reverse
|
||||||
|
.transpose
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
@@ -1,3 +1,3 @@
|
|||||||
module Uplot
|
module Uplot
|
||||||
VERSION = '0.1.0'.freeze
|
VERSION = '0.2.1'.freeze
|
||||||
end
|
end
|
||||||
|
152
test/fixtures/iris.csv
vendored
Normal file
152
test/fixtures/iris.csv
vendored
Normal file
@@ -0,0 +1,152 @@
|
|||||||
|
sepal_length,sepal_width,petal_length,petal_width,species
|
||||||
|
5.1,3.5,1.4,0.2,Iris-setosa
|
||||||
|
4.9,3.0,1.4,0.2,Iris-setosa
|
||||||
|
4.7,3.2,1.3,0.2,Iris-setosa
|
||||||
|
4.6,3.1,1.5,0.2,Iris-setosa
|
||||||
|
5.0,3.6,1.4,0.2,Iris-setosa
|
||||||
|
5.4,3.9,1.7,0.4,Iris-setosa
|
||||||
|
4.6,3.4,1.4,0.3,Iris-setosa
|
||||||
|
5.0,3.4,1.5,0.2,Iris-setosa
|
||||||
|
4.4,2.9,1.4,0.2,Iris-setosa
|
||||||
|
4.9,3.1,1.5,0.1,Iris-setosa
|
||||||
|
5.4,3.7,1.5,0.2,Iris-setosa
|
||||||
|
4.8,3.4,1.6,0.2,Iris-setosa
|
||||||
|
4.8,3.0,1.4,0.1,Iris-setosa
|
||||||
|
4.3,3.0,1.1,0.1,Iris-setosa
|
||||||
|
5.8,4.0,1.2,0.2,Iris-setosa
|
||||||
|
5.7,4.4,1.5,0.4,Iris-setosa
|
||||||
|
5.4,3.9,1.3,0.4,Iris-setosa
|
||||||
|
5.1,3.5,1.4,0.3,Iris-setosa
|
||||||
|
5.7,3.8,1.7,0.3,Iris-setosa
|
||||||
|
5.1,3.8,1.5,0.3,Iris-setosa
|
||||||
|
5.4,3.4,1.7,0.2,Iris-setosa
|
||||||
|
5.1,3.7,1.5,0.4,Iris-setosa
|
||||||
|
4.6,3.6,1.0,0.2,Iris-setosa
|
||||||
|
5.1,3.3,1.7,0.5,Iris-setosa
|
||||||
|
4.8,3.4,1.9,0.2,Iris-setosa
|
||||||
|
5.0,3.0,1.6,0.2,Iris-setosa
|
||||||
|
5.0,3.4,1.6,0.4,Iris-setosa
|
||||||
|
5.2,3.5,1.5,0.2,Iris-setosa
|
||||||
|
5.2,3.4,1.4,0.2,Iris-setosa
|
||||||
|
4.7,3.2,1.6,0.2,Iris-setosa
|
||||||
|
4.8,3.1,1.6,0.2,Iris-setosa
|
||||||
|
5.4,3.4,1.5,0.4,Iris-setosa
|
||||||
|
5.2,4.1,1.5,0.1,Iris-setosa
|
||||||
|
5.5,4.2,1.4,0.2,Iris-setosa
|
||||||
|
4.9,3.1,1.5,0.1,Iris-setosa
|
||||||
|
5.0,3.2,1.2,0.2,Iris-setosa
|
||||||
|
5.5,3.5,1.3,0.2,Iris-setosa
|
||||||
|
4.9,3.1,1.5,0.1,Iris-setosa
|
||||||
|
4.4,3.0,1.3,0.2,Iris-setosa
|
||||||
|
5.1,3.4,1.5,0.2,Iris-setosa
|
||||||
|
5.0,3.5,1.3,0.3,Iris-setosa
|
||||||
|
4.5,2.3,1.3,0.3,Iris-setosa
|
||||||
|
4.4,3.2,1.3,0.2,Iris-setosa
|
||||||
|
5.0,3.5,1.6,0.6,Iris-setosa
|
||||||
|
5.1,3.8,1.9,0.4,Iris-setosa
|
||||||
|
4.8,3.0,1.4,0.3,Iris-setosa
|
||||||
|
5.1,3.8,1.6,0.2,Iris-setosa
|
||||||
|
4.6,3.2,1.4,0.2,Iris-setosa
|
||||||
|
5.3,3.7,1.5,0.2,Iris-setosa
|
||||||
|
5.0,3.3,1.4,0.2,Iris-setosa
|
||||||
|
7.0,3.2,4.7,1.4,Iris-versicolor
|
||||||
|
6.4,3.2,4.5,1.5,Iris-versicolor
|
||||||
|
6.9,3.1,4.9,1.5,Iris-versicolor
|
||||||
|
5.5,2.3,4.0,1.3,Iris-versicolor
|
||||||
|
6.5,2.8,4.6,1.5,Iris-versicolor
|
||||||
|
5.7,2.8,4.5,1.3,Iris-versicolor
|
||||||
|
6.3,3.3,4.7,1.6,Iris-versicolor
|
||||||
|
4.9,2.4,3.3,1.0,Iris-versicolor
|
||||||
|
6.6,2.9,4.6,1.3,Iris-versicolor
|
||||||
|
5.2,2.7,3.9,1.4,Iris-versicolor
|
||||||
|
5.0,2.0,3.5,1.0,Iris-versicolor
|
||||||
|
5.9,3.0,4.2,1.5,Iris-versicolor
|
||||||
|
6.0,2.2,4.0,1.0,Iris-versicolor
|
||||||
|
6.1,2.9,4.7,1.4,Iris-versicolor
|
||||||
|
5.6,2.9,3.6,1.3,Iris-versicolor
|
||||||
|
6.7,3.1,4.4,1.4,Iris-versicolor
|
||||||
|
5.6,3.0,4.5,1.5,Iris-versicolor
|
||||||
|
5.8,2.7,4.1,1.0,Iris-versicolor
|
||||||
|
6.2,2.2,4.5,1.5,Iris-versicolor
|
||||||
|
5.6,2.5,3.9,1.1,Iris-versicolor
|
||||||
|
5.9,3.2,4.8,1.8,Iris-versicolor
|
||||||
|
6.1,2.8,4.0,1.3,Iris-versicolor
|
||||||
|
6.3,2.5,4.9,1.5,Iris-versicolor
|
||||||
|
6.1,2.8,4.7,1.2,Iris-versicolor
|
||||||
|
6.4,2.9,4.3,1.3,Iris-versicolor
|
||||||
|
6.6,3.0,4.4,1.4,Iris-versicolor
|
||||||
|
6.8,2.8,4.8,1.4,Iris-versicolor
|
||||||
|
6.7,3.0,5.0,1.7,Iris-versicolor
|
||||||
|
6.0,2.9,4.5,1.5,Iris-versicolor
|
||||||
|
5.7,2.6,3.5,1.0,Iris-versicolor
|
||||||
|
5.5,2.4,3.8,1.1,Iris-versicolor
|
||||||
|
5.5,2.4,3.7,1.0,Iris-versicolor
|
||||||
|
5.8,2.7,3.9,1.2,Iris-versicolor
|
||||||
|
6.0,2.7,5.1,1.6,Iris-versicolor
|
||||||
|
5.4,3.0,4.5,1.5,Iris-versicolor
|
||||||
|
6.0,3.4,4.5,1.6,Iris-versicolor
|
||||||
|
6.7,3.1,4.7,1.5,Iris-versicolor
|
||||||
|
6.3,2.3,4.4,1.3,Iris-versicolor
|
||||||
|
5.6,3.0,4.1,1.3,Iris-versicolor
|
||||||
|
5.5,2.5,4.0,1.3,Iris-versicolor
|
||||||
|
5.5,2.6,4.4,1.2,Iris-versicolor
|
||||||
|
6.1,3.0,4.6,1.4,Iris-versicolor
|
||||||
|
5.8,2.6,4.0,1.2,Iris-versicolor
|
||||||
|
5.0,2.3,3.3,1.0,Iris-versicolor
|
||||||
|
5.6,2.7,4.2,1.3,Iris-versicolor
|
||||||
|
5.7,3.0,4.2,1.2,Iris-versicolor
|
||||||
|
5.7,2.9,4.2,1.3,Iris-versicolor
|
||||||
|
6.2,2.9,4.3,1.3,Iris-versicolor
|
||||||
|
5.1,2.5,3.0,1.1,Iris-versicolor
|
||||||
|
5.7,2.8,4.1,1.3,Iris-versicolor
|
||||||
|
6.3,3.3,6.0,2.5,Iris-virginica
|
||||||
|
5.8,2.7,5.1,1.9,Iris-virginica
|
||||||
|
7.1,3.0,5.9,2.1,Iris-virginica
|
||||||
|
6.3,2.9,5.6,1.8,Iris-virginica
|
||||||
|
6.5,3.0,5.8,2.2,Iris-virginica
|
||||||
|
7.6,3.0,6.6,2.1,Iris-virginica
|
||||||
|
4.9,2.5,4.5,1.7,Iris-virginica
|
||||||
|
7.3,2.9,6.3,1.8,Iris-virginica
|
||||||
|
6.7,2.5,5.8,1.8,Iris-virginica
|
||||||
|
7.2,3.6,6.1,2.5,Iris-virginica
|
||||||
|
6.5,3.2,5.1,2.0,Iris-virginica
|
||||||
|
6.4,2.7,5.3,1.9,Iris-virginica
|
||||||
|
6.8,3.0,5.5,2.1,Iris-virginica
|
||||||
|
5.7,2.5,5.0,2.0,Iris-virginica
|
||||||
|
5.8,2.8,5.1,2.4,Iris-virginica
|
||||||
|
6.4,3.2,5.3,2.3,Iris-virginica
|
||||||
|
6.5,3.0,5.5,1.8,Iris-virginica
|
||||||
|
7.7,3.8,6.7,2.2,Iris-virginica
|
||||||
|
7.7,2.6,6.9,2.3,Iris-virginica
|
||||||
|
6.0,2.2,5.0,1.5,Iris-virginica
|
||||||
|
6.9,3.2,5.7,2.3,Iris-virginica
|
||||||
|
5.6,2.8,4.9,2.0,Iris-virginica
|
||||||
|
7.7,2.8,6.7,2.0,Iris-virginica
|
||||||
|
6.3,2.7,4.9,1.8,Iris-virginica
|
||||||
|
6.7,3.3,5.7,2.1,Iris-virginica
|
||||||
|
7.2,3.2,6.0,1.8,Iris-virginica
|
||||||
|
6.2,2.8,4.8,1.8,Iris-virginica
|
||||||
|
6.1,3.0,4.9,1.8,Iris-virginica
|
||||||
|
6.4,2.8,5.6,2.1,Iris-virginica
|
||||||
|
7.2,3.0,5.8,1.6,Iris-virginica
|
||||||
|
7.4,2.8,6.1,1.9,Iris-virginica
|
||||||
|
7.9,3.8,6.4,2.0,Iris-virginica
|
||||||
|
6.4,2.8,5.6,2.2,Iris-virginica
|
||||||
|
6.3,2.8,5.1,1.5,Iris-virginica
|
||||||
|
6.1,2.6,5.6,1.4,Iris-virginica
|
||||||
|
7.7,3.0,6.1,2.3,Iris-virginica
|
||||||
|
6.3,3.4,5.6,2.4,Iris-virginica
|
||||||
|
6.4,3.1,5.5,1.8,Iris-virginica
|
||||||
|
6.0,3.0,4.8,1.8,Iris-virginica
|
||||||
|
6.9,3.1,5.4,2.1,Iris-virginica
|
||||||
|
6.7,3.1,5.6,2.4,Iris-virginica
|
||||||
|
6.9,3.1,5.1,2.3,Iris-virginica
|
||||||
|
5.8,2.7,5.1,1.9,Iris-virginica
|
||||||
|
6.8,3.2,5.9,2.3,Iris-virginica
|
||||||
|
6.7,3.3,5.7,2.5,Iris-virginica
|
||||||
|
6.7,3.0,5.2,2.3,Iris-virginica
|
||||||
|
6.3,2.5,5.0,1.9,Iris-virginica
|
||||||
|
6.5,3.0,5.2,2.0,Iris-virginica
|
||||||
|
6.2,3.4,5.4,2.3,Iris-virginica
|
||||||
|
5.9,3.0,5.1,1.8,Iris-virginica
|
||||||
|
|
|
@@ -23,5 +23,6 @@ Gem::Specification.new do |spec|
|
|||||||
spec.add_runtime_dependency 'unicode_plot'
|
spec.add_runtime_dependency 'unicode_plot'
|
||||||
spec.add_development_dependency 'bundler'
|
spec.add_development_dependency 'bundler'
|
||||||
spec.add_development_dependency 'rake'
|
spec.add_development_dependency 'rake'
|
||||||
|
spec.add_development_dependency 'rubocop'
|
||||||
spec.add_development_dependency 'test-unit'
|
spec.add_development_dependency 'test-unit'
|
||||||
end
|
end
|
||||||
|
Reference in New Issue
Block a user