Module: Enumerable

Defined in:
lib/viral_seq/enumerable.rb

Overview

additional statistic/math functions to Module::Enumerable

Examples:

median number

array = [1,2,3,4,5,6,7,8,9,10]
array.median
=> 5.5

sum

array = [1,2,3,4,5,6,7,8,9,10]
array.sum
=> 55

average number (mean)

array = [1,2,3,4,5,6,7,8,9,10]
array.mean
=> 5.5

sample variance

array = [1,2,3,4,5,6,7,8,9,10]
array.sample_variance
=> 9.166666666666666

standard deviation

array = [1,2,3,4,5,6,7,8,9,10]
array.stdev
=> 3.0276503540974917

upper quartile

array = [1,2,3,4,5,6,7,8,9,10]
array.upper_quartile
=> 7.5

lower_quartile

array = [1,2,3,4,5,6,7,8,9,10]
array.lower_quartile
=> 3.5

count frequency of elements in an array

array = %w{cat dog monkey cat cat cat monkey}
array.count_freq
=> {"cat"=>4, "dog"=>1, "monkey"=>2}

count frequency as percentage of elements in an array

array = %w{cat dog monkey cat cat cat monkey}
array.count_freq2
=> {"cat"=>0.57, "dog"=>0.14, "monkey"=>0.29}

Instance Method Summary collapse

Instance Method Details

#count_freq ⇒ Object

tabulate elements and frequencies of an Enumerable return [Hash] return a hash of :element => :freq_count



107
108
109
110
111
112
113
 
# File 'lib/viral_seq/enumerable.rb', line 107

def count_freq
  hash = Hash.new(0)
  self.each do |element|
    hash[element] +=1
  end
  return hash
end

#count_freq2(decimal = 2) ⇒ Object

tabulate elements and frequencies (as percentage) of an Enumerable { return [Hash] return a hash of :element => :percentage

Parameters:

  • decimal (Integer) (defaults to: 2) —

    decimals of frequency



119
120
121
122
123
124
125
126
127
128
129
130
 
# File 'lib/viral_seq/enumerable.rb', line 119

def count_freq2(decimal = 2)
  hash1 = Hash.new(0)
  self.each do |element|
    hash1[element] += 1
  end
  total_elements = self.size
  hash2 = Hash.new(0)
  hash1.each do |key,value|
    hash2[key] = (value/total_elements.to_f).round(decimal)
  end
  return hash2
end

#lower_quartile ⇒ Numeric

generate lower quartile value

Returns:

  • (Numeric) —

    lower quartile value



91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/viral_seq/enumerable.rb', line 91

def lower_quartile
  return nil if self.empty?
  sorted_array = self.sort
  u = 0.25*sorted_array.length + 1
  if (u-u.truncate).is_a?(Integer)
    return sorted_array[(u-u.truncate)-1]
  else
    sample = sorted_array[u.truncate.abs-1]
    sample1 = sorted_array[(u.truncate.abs)]
    return sample+((sample1-sample)*(u-u.truncate))
  end
end

#mean ⇒ Float

generate mean number

Returns:

  • (Float) —

    mean value



56
57
58
 
# File 'lib/viral_seq/enumerable.rb', line 56

def mean
  self.sum/self.length.to_f
end

#median ⇒ Numeric

generate median number

Returns:

  • (Numeric) —

    median number



42
43
44
45
46
 
# File 'lib/viral_seq/enumerable.rb', line 42

def median
  len = self.length
  sorted = self.sort
  len % 2 == 1 ? sorted[len/2] : (sorted[len/2 - 1] + sorted[len/2]).to_f / 2
end

#sample_variance ⇒ Float

generate sample variance

Returns:

  • (Float) —

    sample variance



62
63
64
65
66
 
# File 'lib/viral_seq/enumerable.rb', line 62

def sample_variance
  m = self.mean
  sum = self.inject(0){|accum, i| accum + (i-m)**2 }
  sum/(self.length - 1).to_f
end

#stdev ⇒ Float

generate standard deviation

Returns:

  • (Float) —

    standard deviation



70
71
72
 
# File 'lib/viral_seq/enumerable.rb', line 70

def stdev
  return Math.sqrt(self.sample_variance)
end

#sum ⇒ Numeric

generate summed value

Returns:

  • (Numeric) —

    summed value



50
51
52
 
# File 'lib/viral_seq/enumerable.rb', line 50

def sum
   self.inject(0){|accum, i| accum + i }
end

#upper_quartile ⇒ Numeric

generate upper quartile value

Returns:

  • (Numeric) —

    upper quartile value



76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/viral_seq/enumerable.rb', line 76

def upper_quartile
  return nil if self.empty?
  sorted_array = self.sort
  u = (0.25*(3*sorted_array.length))
  if (u-u.truncate).is_a?(Integer)
    return sorted_array[(u-u.truncate)-1]
  else
    sample = sorted_array[u.truncate.abs-1]
    sample1 = sorted_array[(u.truncate.abs)]
    return sample+((sample1-sample)*(u-u.truncate))
  end
end