Merge pull request #236 from Joshua-Dias-Barreto/numericalColumn

#numericalColumns now returns an array of all numerical columns

Author: jecisc

src/DataFrame-Tests/DataFrameTest.class.st

@@ -2841,18 +2841,17 @@ DataFrameTest >> testNumericalColumnNames [
 { #category : #tests }
 DataFrameTest >> testNumericalColumns [
 
-	| dataFrame expected |
+	| dataFrame expectedCollection |
 	dataFrame := DataFrame withRows:
 		             #( #( Male 21 Argentina 94 ) #( Female 20 France 97 )
 		                #( Female 22 Spain 95 ) #( Male 24 Portugal 99 ) ).
 	dataFrame columnNames: #( Gender Age Country Score ).
 
-	expected := DataFrame withRows:
-		            #( #( 21 94 ) #( 20 97 ) #( 22 95 ) #( 24 99 ) ).
-
-	expected columnNames: #( Age Score ).
+	expectedCollection := {
+		                      (dataFrame columnAt: 2).
+		                      (dataFrame columnAt: 4) } asArray.
 
-	self assert: dataFrame numericalColumns equals: expected
+	self assert: dataFrame numericalColumns equals: expectedCollection
 ]
 
 { #category : #splitjoin }

src/DataFrame/DataFrame.class.st

@@ -1439,7 +1439,8 @@ DataFrame >> numericalColumnNames [
 DataFrame >> numericalColumns [
 	"Returns all numerical columns of the dataframe"
 
-	^ self columns: self numericalColumnNames
+	^ self columns select: [ :column |
+		  (self dataTypes at: column name) includesBehavior: Number ]
 ]
 
 { #category : #splitjoin }