kotlin-numpy / org.jetbrains.numkt.core / argSort

argSort

fun <T : Any> KtNDArray<T>.argSort(axis: Int? = -1, kind: String? = null): KtNDArray<Long> (source)

Returns the indices that would sort this array.

Parameters

axis - axis along which to sort. The default is -1 (the last axis). If null, the flattened array is used.

kind - sorting algorithm from {'quicksort', 'mergesort', 'heapsort', 'stable'}. The default is 'quicksort'.

Return
KtNDArray of Long type. Array of indices that sort this array along the specified axis.