Added #sortByRowNamesUsing: with tests.

Author: Joshua-Dias-Barreto

src/DataFrame-Tests/DataFrameTest.class.st

@@ -5157,6 +5157,31 @@ DataFrameTest >> testSortByAll [
 	self assert: actual equals: expected
 ]
 
+{ #category : #tests }
+DataFrameTest >> testSortByRowNamesUsing [
+	"sorts by second letter of row name"
+
+	| dataFrame expected sortedDF |
+	dataFrame := DataFrame withRows:
+		             #( #( 'CD' 3 2 ) 
+							 #( 'AC' 1 4 ) 
+							 #( 'DB' 2 4 )
+		                #( 'BA' 5 1 ) ).
+	dataFrame rowNames: #( 'CD' 'AC' 'DB' 'BA' ).
+
+	expected := DataFrame withRows:
+		            #( #( 'BA' 5 1 ) 
+							#( 'DB' 2 4 ) 
+							#( 'AC' 1 4 )
+		               #( 'CD' 3 2 ) ).
+	expected rowNames: #( 'BA' 'DB' 'AC' 'CD' ).
+
+	sortedDF := dataFrame sortByRowNamesUsing: [ :name1 :name2 |
+		            name1 second <= name2 second ].
+
+	self assert: sortedDF equals: expected
+]
+
 { #category : #tests }
 DataFrameTest >> testSortByUsing [
 	"Sort by second letter of city name"

src/DataFrame/DataFrame.class.st

@@ -2469,6 +2469,22 @@ DataFrame >> sortByAll: arrayOfColumnNames [
 	^ self
 ]
 
+{ #category : #sorting }
+DataFrame >> sortByRowNamesUsing: aBlock [
+	"Sorts the rows of the data frame based on the row names using the given comparison block"
+
+	| sortedKeys newContents |
+	sortedKeys := self rowNames sorted: aBlock.
+
+	newContents := DataFrameInternal new: self dimensions.
+
+	sortedKeys withIndexDo: [ :key :i |
+		newContents rowAt: i put: (self row: key) asArray ].
+
+	contents := newContents.
+	self rowNames: sortedKeys
+]
+
 { #category : #sorting }
 DataFrame >> sortDescendingBy: columnName [
 	"Rearranges the rows of the data frame in descending order of the values in the column named columnName"