#median can now handle nils. Used removeNils for the implementation of the other methods.

Author: Joshua-Dias-Barreto

src/DataFrame-Tests/DataSeriesTest.class.st

@@ -1600,6 +1600,18 @@ DataSeriesTest >> testMaxWithNils [
 	self assert: #( 1 nil 3 nil ) asDataSeries max equals: 3
 ]
 
+{ #category : #tests }
+DataSeriesTest >> testMedian [
+
+	self assert: #( 1 2 3 4 5 ) asDataSeries median equals: 3
+]
+
+{ #category : #tests }
+DataSeriesTest >> testMedianWithNils [
+
+	self assert: #( 1 2 nil 3 4 nil 5 ) asDataSeries median equals: 3
+]
+
 { #category : #'tests - statistics' }
 DataSeriesTest >> testMin [
 

src/DataFrame/DataSeries.class.st

@@ -167,11 +167,11 @@ DataSeries >> atIndex: aNumber transform: aBlock [
 	self at: key transform: aBlock
 ]
 
-{ #category : #information }
+{ #category : #statistics }
 DataSeries >> average [
-	"We do not count the nils"
+	"Returns the average without including nils"
 
-	^ (self values reject: #isNil) average
+	^ self removeNils values average
 ]
 
 { #category : #'data-types' }
@@ -524,14 +524,21 @@ DataSeries >> makeNumerical [
 DataSeries >> max [
 	"Returns the maximum value of the dataseries without including nils"
 
-	^ (self values reject: #isNil) max
+	^ self removeNils values max
+]
+
+{ #category : #statistics }
+DataSeries >> median [
+	"Returns the median without including nils"
+
+	^ self removeNils values median
 ]
 
 { #category : #statistics }
 DataSeries >> min [
 	"Returns the minimum value of the dataseries without including nils"
 
-	^ (self values reject: #isNil) min
+	^ self removeNils values min
 ]
 
 { #category : #accessing }
@@ -775,7 +782,7 @@ DataSeries >> sortedDescending [
 DataSeries >> stdev [
 	"Returns the standard deviation of the dataseries without including nils"
 
-	^ (self values reject: #isNil) stdev
+	^ self removeNils values stdev
 ]
 
 { #category : #transformation }