jtescher/descriptive-statistics

module DescriptiveStatistics
  module Spread
    def variance
      return if length < 1
      precalculated_mean = mean
      sum = self.inject(0) {|accumulator, value| accumulator + (value - precalculated_mean) ** 2 }
      sum / (length.to_f - 1)
    end

    def population_variance
      return if length < 1
      precalculated_mean = mean
      sum = self.inject(0) {|accumulator, value| accumulator + (value - precalculated_mean) ** 2 }
      sum / length.to_f
    end

    def standard_deviation
      return if length < 2
      Math.sqrt(variance)
    end

    def relative_standard_deviation
      return if length < 1
      precalculated_mean = mean
      (population_standard_deviation / precalculated_mean) * 100.0
    end

    def population_standard_deviation
      return if length < 1
      Math.sqrt(population_variance)
    end

    def zscore
      return if length < 2
      stdev = standard_deviation
      m = mean
      stdev.zero? ? Array.new(self.length, 0) : self.collect { |v| (v - m) / stdev }
    end
  end
end