Module: Daru::Plotly::Vector

Extended by:
Daru::Plotly
Defined in:
lib/daru/plotly/vector.rb

Overview

exports #plot and #generate_data for Vector

Constant Summary

Constants included from Daru::Plotly

VERSION

Class Method Summary collapse

Methods included from Daru::Plotly

default_layout, extract_mode_string, extract_type, plot

Class Method Details

.generate_data(vector, opts) ⇒ Object



11
12
13
14
15
16
17
18
19
20
21
22
# File 'lib/daru/plotly/vector.rb', line 11

def generate_data(vector, opts)
  type = extract_type opts[:type]

  case type
  when :histogram
    generate_data_for_histogram(vector, opts)
  when :scatter, :bar
    generate_data_for_scatter_and_bar(vector, opts)
  when :pie
    generate_data_for_pie(vector, opts)
  end.map { |trace| trace.merge(opts[:opts] || {}) }
end

.generate_data_for_histogram(vector, _opts) ⇒ Object



24
25
26
# File 'lib/daru/plotly/vector.rb', line 24

def generate_data_for_histogram(vector, _opts)
  [{ x: vector.to_a, type: :histogram }]
end

.generate_data_for_pie(vector, _opts) ⇒ Object



37
38
39
# File 'lib/daru/plotly/vector.rb', line 37

def generate_data_for_pie(vector, _opts)
  [{ labels: vector.index.to_a, values: vector.to_a, type: :pie }]
end

.generate_data_for_scatter_and_bar(vector, opts) ⇒ Object



28
29
30
31
32
33
34
35
# File 'lib/daru/plotly/vector.rb', line 28

def generate_data_for_scatter_and_bar(vector, opts)
  [
    {
      x: vector.index.to_a, y: vector.to_a,
      type: opts[:type], mode: extract_mode_string(opts[:model])
    }
  ]
end

.supported_typesObject



41
42
43
# File 'lib/daru/plotly/vector.rb', line 41

def supported_types
  i[scatter bar histogram pie]
end