Assume, we would like to sort the rows of a 2d matrix by the the last column (representing "age"). This can be done with
// sort by last column sorted = matrix.viewSorted(matrix.columns()-1);
Or assume, we would like to sort the columns of a 2d matrix by the the last row. Unfortunately, there is no convenience method to directly sort by row. So we need to view columns as rows and rows as columns, then sort, then adjust our view again.
// sort by last row int lastRow = matrix.rows()-1; sorted = matrix.viewDice().viewSorted(lastRow).viewDice();
Next, we would like to sort the rows of a 2d matrix by the aggregate sum of values in a row. A comparator object is used to do the job:
// sort by sum of values in a row
DoubleMatrix1DComparator comp = new DoubleMatrix1DComparator() {
public int compare(DoubleMatrix1D a, DoubleMatrix1D b) {
double as = a.zSum(); double bs = b.zSum();
return as < bs ? -1 : as == bs ? 0 : 1;
}
};
sorted = cern.colt.matrix.doublealgo.Sorting.quickSort(matrix,comp);
Further, we would like to sort the rows of a 2d matrix by the aggregate sum of logarithms in a row (which is a way to achieve sorting by geometric mean when viewing a row as a series of samples). A slightly more complex comparator object is needed:
// sort by sum of logarithms in a row
DoubleMatrix1DComparator comp = new DoubleMatrix1DComparator() {
public int compare(DoubleMatrix1D a, DoubleMatrix1D b) {
double as = a.aggregate(cern.jet.math.Functions.plus,cern.jet.math.Functions.log);
double bs = b.aggregate(cern.jet.math.Functions.plus,cern.jet.math.Functions.log);
return as < bs ? -1 : as == bs ? 0 : 1;
}
};
sorted = cern.colt.matrix.doublealgo.Sorting.quickSort(matrix,comp);
This is certainly not most efficient since row sums are recomputed many times
(2*rows*log(rows) times, on average), but will suffice as an example.
An efficient app will precompute the sums and use cern.colt.GenericSorting
to sort the matrix. In general, if comparisons are expensive, precomputation boots
performance by a factor 2*log(rows).
Recently,
two methods that do exactly that were added to cern.colt.matrix.doublealgo.Sorting.
One of them works by filling a row into a so-called "bin", which is a multi-set
with statistics operations defined upon. Aggregate measures over the row are
then computed via a BinFunction1D.
Some prefabricated functions are contained in BinFunctions1D
Here is how to solve the problem efficiently:
// sort by sum of logarithms in a row sorted = cern.colt.matrix.doublealgo.Sorting.quickSort(matrix,hep.aida.bin.BinFunctions1D.sumLog); // sort by median in a row sorted = cern.colt.matrix.doublealgo.Sorting.quickSort(matrix,hep.aida.bin.BinFunctions1D.median); // sort by maximum in a row sorted = cern.colt.matrix.doublealgo.Sorting.quickSort(matrix,hep.aida.bin.BinFunctions1D.max);