Class: Statsample::Graph::Scatterplot

Inherits:
Object
  • Object
show all
Includes:
Summarizable
Defined in:
lib/statsample/graph/scatterplot.rb

Overview

Scatterplot

From Wikipedia: A scatter plot or scattergraph is a type of mathematical diagram using Cartesian coordinates to display values for two variables for a set of data.

The data is displayed as a collection of points, each having the value of one variable determining the position on the horizontal axis and the value of the other variable determining the position on the vertical axis. This kind of plot is also called a scatter chart, scatter diagram and scatter graph.

Usage

Svg output

a = Daru::Vector.new([1,2,3,4])
b = Daru::Vector.new([3,4,5,6])
puts Statsample::Graph::Scatterplot.new(a,b).to_svg

Using ReportBuilder

a = Daru::Vector.new([1,2,3,4])
b = Daru::Vector.new([3,4,5,6])
rb=ReportBuilder.new
rb.add(Statsample::Graph::Scatterplot.new(a,b))
rb.save_html('scatter.html')

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Summarizable

#summary

Constructor Details

#initialize(v1, v2, opts = Hash.new) ⇒ Scatterplot

Create a new Scatterplot. Params:

  • v1: Vector on X axis

  • v2: Vector on Y axis

  • opts: Hash of options. See attributes of Scatterplot

[View source] [View on GitHub]

68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/statsample/graph/scatterplot.rb', line 68

def initialize(v1,v2,opts=Hash.new)
  @v1_name,@v2_name = v1.name,v2.name
  @v1,@v2           = Statsample.only_valid_clone(v1,v2)
  opts_default={
    :name=>_("Scatterplot (%s - %s)") % [@v1_name, @v2_name],
    :width=>400,
    :height=>300,
    :dot_alpha=>0.5,
    :line_median=>false,
    :margin_top=>10,
    :margin_bottom=>20,
    :margin_left=>20,
    :margin_right=>20,
    :minimum_x=>nil,
    :maximum_x=>nil,
    :minimum_y=>nil,
    :maximum_y=>nil,
    :groups=>nil
  }
  @opts=opts_default.merge(opts)
  opts_default.keys.each {|k| send("#{k}=", @opts[k]) }
  @data=[]
  @v1.each_with_index {|d1,i|
    @data.push({:x=>d1, :y=>@v2[i]})
  }
end

Instance Attribute Details

#data ⇒ Object (readonly)

Returns the value of attribute data.

[View on GitHub]

42
43
44
 
# File 'lib/statsample/graph/scatterplot.rb', line 42

def data
  @data
end

#dot_alpha ⇒ Object

Returns the value of attribute dot_alpha.

[View on GitHub]

30
31
32
 
# File 'lib/statsample/graph/scatterplot.rb', line 30

def dot_alpha
  @dot_alpha
end

#groups ⇒ Object

Array with assignation to groups of bars For example, for four vectors, 

boxplot.groups=[1,2,1,3]

Assign same color to first and third element, and different to second and fourth

[View on GitHub]

50
51
52
 
# File 'lib/statsample/graph/scatterplot.rb', line 50

def groups
  @groups
end

#height ⇒ Object

Total height of Scatterplot

[View on GitHub]

29
30
31
 
# File 'lib/statsample/graph/scatterplot.rb', line 29

def height
  @height
end

#line_median ⇒ Object

Add a line on median of x and y axis

[View on GitHub]

32
33
34
 
# File 'lib/statsample/graph/scatterplot.rb', line 32

def line_median
  @line_median
end

#margin_bottom ⇒ Object

Bottom margin

[View on GitHub]

36
37
38
 
# File 'lib/statsample/graph/scatterplot.rb', line 36

def margin_bottom
  @margin_bottom
end

#margin_left ⇒ Object

Left margin

[View on GitHub]

38
39
40
 
# File 'lib/statsample/graph/scatterplot.rb', line 38

def margin_left
  @margin_left
end

#margin_right ⇒ Object

Right margin

[View on GitHub]

40
41
42
 
# File 'lib/statsample/graph/scatterplot.rb', line 40

def margin_right
  @margin_right
end

#margin_top ⇒ Object

Top margin

[View on GitHub]

34
35
36
 
# File 'lib/statsample/graph/scatterplot.rb', line 34

def margin_top
  @margin_top
end

#maximum_x ⇒ Object

Maximum value on x axis. Calculated automaticly from data if not set

[View on GitHub]

57
58
59
 
# File 'lib/statsample/graph/scatterplot.rb', line 57

def maximum_x
  @maximum_x
end

#maximum_y ⇒ Object

Maximum value on y axis. Calculated automaticly from data if not set.

[View on GitHub]

61
62
63
 
# File 'lib/statsample/graph/scatterplot.rb', line 61

def maximum_y
  @maximum_y
end

#minimum_x ⇒ Object

Minimum value on x axis. Calculated automaticly from data if not set

[View on GitHub]

55
56
57
 
# File 'lib/statsample/graph/scatterplot.rb', line 55

def minimum_x
  @minimum_x
end

#minimum_y ⇒ Object

Minimum value on y axis. Set to 0 if not set

[View on GitHub]

59
60
61
 
# File 'lib/statsample/graph/scatterplot.rb', line 59

def minimum_y
  @minimum_y
end

#name ⇒ Object

Returns the value of attribute name.

[View on GitHub]

25
26
27
 
# File 'lib/statsample/graph/scatterplot.rb', line 25

def name
  @name
end

#v1 ⇒ Object (readonly)

Returns the value of attribute v1.

[View on GitHub]

43
44
45
 
# File 'lib/statsample/graph/scatterplot.rb', line 43

def v1
  @v1
end

#v2 ⇒ Object (readonly)

Returns the value of attribute v2.

[View on GitHub]

43
44
45
 
# File 'lib/statsample/graph/scatterplot.rb', line 43

def v2
  @v2
end

#width ⇒ Object

Total width of Scatterplot

[View on GitHub]

27
28
29
 
# File 'lib/statsample/graph/scatterplot.rb', line 27

def width
  @width
end

#x_scale ⇒ Object (readonly)

Returns the value of attribute x_scale.

[View on GitHub]

53
54
55
 
# File 'lib/statsample/graph/scatterplot.rb', line 53

def x_scale
  @x_scale
end

#y_scale ⇒ Object (readonly)

Returns the value of attribute y_scale.

[View on GitHub]

53
54
55
 
# File 'lib/statsample/graph/scatterplot.rb', line 53

def y_scale
  @y_scale
end

Instance Method Details

#add_line_median(vis) ⇒ Object

Add a rule on median of X and Y axis

[View source] [View on GitHub]

95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/statsample/graph/scatterplot.rb', line 95

def add_line_median(vis) # :nodoc:
  that=self
  x=@x_scale
  y=@y_scale
  vis.execute {
    rule do
      data [that.v1.median]
      left x
      stroke_style Rubyvis.color("#933").alpha(0.5)
      label(:anchor=>"top") do
        text x.tick_format
      end
    end
    rule do
      data [that.v2.median]
      bottom y
      stroke_style Rubyvis.color("#933").alpha(0.5)
      label(:anchor=>"right") do
        text y.tick_format
      end
    end  
  }
  
end

#report_building(builder) ⇒ Object

:nodoc:

[View source] [View on GitHub]

206
207
208
209
210
 
# File 'lib/statsample/graph/scatterplot.rb', line 206

def report_building(builder) # :nodoc:
  builder.section(:name=>name) do |b|
    b.image(to_svg, :type=>'svg', :width=>width, :height=>height)
  end       
end

#rubyvis_panel ⇒ Object

Returns a Rubyvis panel with scatterplot

[View source] [View on GitHub]

120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
 
# File 'lib/statsample/graph/scatterplot.rb', line 120

def rubyvis_panel # :nodoc:
  that=self
  #p @v1.map {|v| v}
  
  @minimum_x||=@v1.min
  @maximum_x||=@v1.max
  @minimum_y||=@v2.min
  @maximum_y||=@v2.max
  
  colors=Rubyvis::Colors.category10
  
  margin_hor=margin_left + margin_right
  margin_vert=margin_top  + margin_bottom
  
  x=Rubyvis::Scale.linear(@minimum_x, @maximum_x).range(0, width - margin_hor)
  y=Rubyvis::Scale.linear(@minimum_y, @maximum_y).range(0, height - margin_vert)
  @x_scale=x
  @y_scale=y
  vis=Rubyvis::Panel.new do |pan| 
    pan.width  width  - margin_hor
    pan.height height - margin_vert
    pan.bottom margin_bottom
    pan.left   margin_left
    pan.right  margin_right
    pan.top    margin_top
    # X axis
    pan.rule do
      data y.ticks
      bottom y
      stroke_style {|d| d!=0 ? "#eee" : "#000"}
      label(:anchor=>'left') do
        visible {|d| d!=0 and  d < that.width}
        text y.tick_format
      end
    end
    
    # Y axis
    pan.rule do
      data x.ticks
      left x
      stroke_style {|d| d!=0 ? "#eee" : "#000"}
      label(:anchor=>'bottom') do
        visible {|d| d>0 and d < that.height}
        text x.tick_format
      end
    end
    # Add lines on median
    add_line_median(pan) if line_median

    pan.panel do
      data(that.data)
      dot do
        left   {|d| x[d[:x]]}
        bottom {|d| y[d[:y]]}
        
        fill_style {|v| 
          alpha=(that.dot_alpha-0.3<=0) ? 0.1 : that.dot_alpha-0.3
          if that.groups
            
            colors.scale(that.groups[index]).alpha(alpha)
          else
            colors.scale(0).alpha(alpha)
          end
        }
        
        stroke_style {|v|
          if that.groups
            colors.scale(that.groups[parent.index]).alpha(that.dot_alpha)
          else
            colors.scale(0).alpha(that.dot_alpha)
          end
        }
        shape_radius 2
      end
    end
  end
  vis
end

#to_svg ⇒ Object

Returns SVG with scatterplot

[View source] [View on GitHub]

200
201
202
203
204
 
# File 'lib/statsample/graph/scatterplot.rb', line 200

def to_svg
  rp = rubyvis_panel
  rp.render
  rp.to_svg
end