mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-05-07 23:51:12 +08:00
Add the Processing module (tentative)
This commit is contained in:
parent
9849898cb1
commit
f2bd99ed2e
24
lib/youplot/backends/processing.rb
Normal file
24
lib/youplot/backends/processing.rb
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
module YouPlot
|
||||||
|
# plotting functions.
|
||||||
|
module Backends
|
||||||
|
module Processing
|
||||||
|
module_function
|
||||||
|
|
||||||
|
def count_values(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
|
||||||
|
end
|
@ -1,5 +1,6 @@
|
|||||||
# frozen_string_literal: true
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
require_relative 'processing'
|
||||||
require 'unicode_plot'
|
require 'unicode_plot'
|
||||||
|
|
||||||
module YouPlot
|
module YouPlot
|
||||||
@ -13,7 +14,7 @@ module YouPlot
|
|||||||
series = data.series
|
series = data.series
|
||||||
# `uplot count`
|
# `uplot count`
|
||||||
if count
|
if count
|
||||||
series = Preprocessing.count_values(series[0])
|
series = Processing.count_values(series[0])
|
||||||
params.title = headers[0] if headers
|
params.title = headers[0] if headers
|
||||||
end
|
end
|
||||||
if series.size == 1
|
if series.size == 1
|
||||||
|
11
test/youplot/backends/processing_test.rb
Normal file
11
test/youplot/backends/processing_test.rb
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
require_relative '../../test_helper'
|
||||||
|
|
||||||
|
class YouPlotCommandTest < Test::Unit::TestCase
|
||||||
|
test :count_values do
|
||||||
|
@m = YouPlot::Backends::Processing
|
||||||
|
assert_equal([%i[a b c], [3, 2, 1]], @m.count_values(%i[a a a b b c]))
|
||||||
|
assert_equal([%i[c b a], [3, 2, 1]], @m.count_values(%i[a b b c c c]))
|
||||||
|
end
|
||||||
|
end
|
Loading…
Reference in New Issue
Block a user