Module: Daru::Plotting::DataFrame

Included in:
DataFrame
Defined in:
lib/daru/plotting/dataframe.rb

Instance Method Summary collapse

Instance Method Details

#plot(opts = {}) {|plot, diagram| ... } ⇒ Object

Plots a DataFrame with Nyaplot on IRuby using the given options. Yields the corresponding Nyaplot::Plot object and the Nyaplot::Diagram object to the block, if it is specified. See the nyaplot docs for info on how to further use these objects.

Detailed instructions on use of the plotting API can be found in the notebooks whose links you can find in the README.

Options

  • :type - Type of plot. Can be :scatter, :bar, :histogram, :line or :box.

  • :x - Vector to be used for X co-ordinates.

  • :y - Vector to be used for Y co-ordinates.

Usage

# Simple bar chart
df = Daru::DataFrame.new({a:['A', 'B', 'C', 'D', 'E'], b:[10,20,30,40,50]})
df.plot type: :bar, x: :a, y: :b

Yields:



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/daru/plotting/dataframe.rb', line 22

def plot opts={}
  options = {
    type:  :scatter
  }.merge(opts)

  plot = Nyaplot::Plot.new
  types = extract_option :type, options

  diagram =
  case 
  when !([:scatter, :bar, :line, :histogram] & types).empty?
    if single_diagram? options
      add_single_diagram plot, options
    else
      add_multiple_diagrams plot, options
    end
  when types.include?(:box)
    numeric = self.only_numerics(clone: false).dup_only_valid

    plot.add_with_df(
      numeric.to_nyaplotdf,
      :box, *numeric.vectors.to_a)
  end

  yield(plot, diagram) if block_given?

  plot.show
end