Merge pull request #230 from Joshua-Dias-Barreto/handlingNils

Added DataFrame>> #replaceNilsWithNextRowValue

Author: jecisc

src/DataFrame-Tests/DataFrameTest.class.st

@@ -3869,6 +3869,30 @@ DataFrameTest >> testReplaceNilsWithMode [
 	self assert: df replaceNilsWithMode equals: expected
 ]
 
+{ #category : #tests }
+DataFrameTest >> testReplaceNilsWithNextRowValue [
+
+	|expected|
+
+	df := DataFrame withRows: #(
+		(7 1 nil)
+   		(8 nil 3)
+   		(nil 2 nil)).
+
+	df rowNames: #(A B C).
+	df columnNames: #(Column1 Column2 Column3).
+
+	expected := DataFrame withRows: #(
+		(7 1 3)
+   		(8 2 3)
+   		(nil 2 nil)).
+
+	expected rowNames: #(A B C).
+	expected columnNames: #(Column1 Column2 Column3).
+
+	self assert: (df replaceNilsWithNextRowValue) equals: expected
+]
+
 { #category : #replacing }
 DataFrameTest >> testReplaceNilsWithPreviousRowValue [
 

src/DataFrame/DataFrame.class.st

@@ -1793,6 +1793,18 @@ DataFrame >> replaceNilsWithMode [
 		modeOfColumn := nil ]
 ]
 
+{ #category : #replacing }
+DataFrame >> replaceNilsWithNextRowValue [
+
+	| value numberOfRows |
+	numberOfRows := self numberOfRows.
+	1 to: self numberOfColumns do: [ :i |
+		self numberOfRows to: 1 by: -1 do: [ :j |
+			j < numberOfRows ifTrue: [
+				(self at: j at: i) ifNil: [ self at: j at: i put: value ] ].
+			value := self at: j at: i ] ]
+]
+
 { #category : #replacing }
 DataFrame >> replaceNilsWithPreviousRowValue [
 	"Replaces all nil values of a data frame with the previous non-nil value of the column in which it is present"