dpnp.ndarray.argsort

method

ndarray.argsort(axis=-1, kind=None, order=None, *, descending=False, stable=None)

Return an ndarray of indices that sort the array along the specified axis.

Refer to dpnp.argsort for full documentation.

Parameters:

axis{None, int}, optional

Axis along which to sort. If None, the array is flattened before sorting. The default is -1, which sorts along the last axis.

Default: -1.

kind{None, "stable", "mergesort", "radixsort"}, optional

Sorting algorithm. The default is None, which uses parallel merge-sort or parallel radix-sort algorithms depending on the array data type.

Default: None.

descendingbool, optional

Sort order. If True, the array must be sorted in descending order (by value). If False, the array must be sorted in ascending order (by value). NaN values (and complex values with a NaN component) are ordered to the end regardless of descending.

Default: False.

stable{None, bool}, optional

Sort stability. If True, the returned array will maintain the relative order of a values which compare as equal. The same behavior applies when set to False or None. Internally, this option selects kind="stable".

Default: None.

See also

dpnp.sort

Return a sorted copy of an array.

dpnp.argsort

Return the indices that would sort an array.

dpnp.lexsort

Indirect stable sort on multiple keys.

dpnp.searchsorted

Find elements in a sorted array.

dpnp.partition

Partial sort.

Examples

>>> import dpnp as np
>>> a = np.array([3, 1, 2])
>>> a.argsort()
array([1, 2, 0])

>>> a = np.array([[0, 3], [2, 2]])
>>> a.argsort(axis=0)
array([[0, 1],
       [1, 0]])