Closed #74. Made it possible to mutate rows with DataFrame>>do:

Author: olekscode

src/DataFrame-Tests/DataFrameTest.class.st

@@ -992,6 +992,24 @@ DataFrameTest >> testDo [
 	self assert: actual equals: expected.
 ]
 
+{ #category : #tests }
+DataFrameTest >> testDoCanModifyRows [
+	| expected |
+
+	expected := DataFrame
+		withRows: #(
+			(BARCELONA 1.609 true)
+			(DUBAI 2.789 true)
+			(LONDON 8.788 false))
+		rowNames: df rowNames
+		columnNames: df columnNames.
+	
+	df do: [ :row |
+		row at: #City put: (row at: #City) asUppercase ].
+	
+	self assert: df equals: expected.
+]
+
 { #category : #tests }
 DataFrameTest >> testEquality [
 
@@ -1586,3 +1604,22 @@ DataFrameTest >> testWithKeyDo [
 
 	self assert: actual equals: expected.
 ]
+
+{ #category : #tests }
+DataFrameTest >> testWithKeyDoCanModifyRows [
+	| expected |
+
+	expected := DataFrame
+		withRows: #(
+			(Barcelona 1.609 true)
+			(DUBAI 2.789 true)
+			(LONDON 8.788 false))
+		rowNames: df rowNames
+		columnNames: df columnNames.
+	
+	df withKeyDo: [ :row :key |
+		row name = 'A'
+			ifFalse: [ row at: #City put: (row at: #City) asUppercase ] ].
+	
+	self assert: df equals: expected.
+]

src/DataFrame/DataFrame.class.st

@@ -647,9 +647,13 @@ DataFrame >> dimensions [
 { #category : #enumerating }
 DataFrame >> do: aBlock [
 "We enumerate through the data enrties - through rows of a data frame"
-
+	| row |
+	
 	1 to: self numberOfRows do: [ :i |
-		aBlock value: (self rowAt: i) ].
+		row := self rowAt: i.
+		aBlock value: row. 
+		"A hack to allow modification of rows inside do block"
+		self rowAt: i put: row asArray ].
 ]
 
 { #category : #statistics }
@@ -1174,8 +1178,14 @@ DataFrame >> variance [
 { #category : #enumerating }
 DataFrame >> withKeyDo: elementAndKeyBlock [
 
-	1 to: self size do: [ :index |
+	1 to: self size do: [ :i |
+		| row |
+		row := (self rowAt: i).
+		
 		elementAndKeyBlock
-			value: (self rowAt: index)
-			value: (rowNames at: index) ].
+			value: row
+			value: (rowNames at: i).
+			
+		"A hack to allow modification of rows inside do block"
+		self rowAt: i put: row asArray ].
 ]