ar_result_calculations 0.0.3 → 0.0.4

data/CHANGELOG

@@ -1,3 +1,7 @@
+ar_result_calculations (0.0.4) December 2, 2010
+
+* Add calculations for median and mode
+
 ar_result_calculations (0.0.3) December 2, 2010
 
 * Add calculations for sample_variance and standard_deviation

data/README.textile

@@ -30,6 +30,7 @@ p. All methods take one parameter (column name).
 |make_numeric|Coerces a column to be numeric.  Useful if you have derived columns returned from a query that are not in the model definition and hence are otherwise returned as strings.  Returns **self** so is composable|
 |sample_variance|Calculates the sample variance of a column (for active record result sets) or self (for an array)|
 |standard_deviation|Calculates the standard deviation of a column (for active record result sets) or self (for an array)|
+|median|Calculates the median of a column (for active record result sets) or self (for an array)|
 
 h1. License
 

data/lib/ar_result_calculations/ar_result_calculations.rb

@@ -107,6 +107,35 @@ module ArResultCalculations
         Math.sqrt(variance)
       end
       
+      # Return the median of a column in an active record result set (or self)
+      #
+      #  column:  The column of which you want the median
+      #  already_sorted: Is the data already sorted (default: true)
+    	def median(column = nil, options = {:already_sorted => true})
+    	  options, column = column, nil if column && column.is_a?(Hash)
+    	  data = column ? map(&column.to_sym) : self
+    	  return nil if data.empty?
+    	  data = data.sort unless options[:already_sorted]
+    	  median_position = length / 2
+    	  length_is_even? ? data[median_position] : mean(data[median_position - 1..median_position])
+    	end
+      
+      # Return the mode(s) of a column in an active record result set (or self)
+      #
+      #  column: The column of which you want the mode (or modes)
+      #  find_all: Find all modes (default: true)
+      def mode(column = nil, options = {:find_all => false})
+    	  options, column = column, nil if column && column.is_a?(Hash)
+        data = column ? map(&column.to_sym) : self
+    	  histogram = data.inject(Hash.new(0)) { |h, n| h[n] += 1; h }
+    	  modes = nil
+    	  histogram.each_pair do |item, times|
+    	    modes << item if modes && times == modes[0] and options[:find_all]
+    	    modes = [times, item] if (!modes && times > 1) or (modes && times > modes[0])
+    	  end
+        modes && options[:find_all] ? modes[1..modes.size] : modes.try(:[], 1)
+    	end
+      	
       # Force a column to be numeric.  Useful if you have derived
       # columns from a query that is not part of the base model.
       # 
@@ -141,6 +170,10 @@ module ArResultCalculations
       def is_float?(val)
         val.is_a?(Float) || val.is_a?(Rational)
       end
+      
+      def length_is_even?
+        length % 2 == 0 
+      end
     end
     
     module ClassMethods

data/lib/ar_result_calculations/version.rb

@@ -1,3 +1,3 @@
 module ArResultCalculations
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: ar_result_calculations
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 23
   prerelease: false
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - Kip Cole