# Sorting, Searching, and Counting¶

cupy.sort(a)[source]

Returns a sorted copy of an array with a stable sorting algorithm.

Parameters: a (cupy.ndarray) – Array to be sorted. Array of the same type and shape as a. cupy.ndarray

Note

For its implementation reason, cupy.sort currently supports only arrays with their rank of one and does not support axis, kind and order parameters that numpy.sort does support.

cupy.argmax(a, axis=None, dtype=None, out=None, keepdims=False)[source]

Returns the indices of the maximum along an axis.

Parameters: a (cupy.ndarray) – Array to take argmax. axis (int) – Along which axis to find the maximum. a is flattened by default. dtype – Data type specifier. out (cupy.ndarray) – Output array. keepdims (bool) – If True, the axis axis is preserved as an axis of length one. The indices of the maximum of a along an axis. cupy.ndarray
cupy.argmin(a, axis=None, dtype=None, out=None, keepdims=False)[source]

Returns the indices of the minimum along an axis.

Parameters: a (cupy.ndarray) – Array to take argmin. axis (int) – Along which axis to find the minimum. a is flattened by default. dtype – Data type specifier. out (cupy.ndarray) – Output array. keepdims (bool) – If True, the axis axis is preserved as an axis of length one. The indices of the minimum of a along an axis. cupy.ndarray
cupy.count_nonzero(a, axis=None)[source]

Counts the number of non-zero values in the array.

Parameters: a (cupy.ndarray) – The array for which to count non-zeros. axis (int or tuple, optional) – Axis or tuple of axes along which to count non-zeros. Default is None, meaning that non-zeros will be counted along a flattened version of a Number of non-zero values in the array along a given axis. Otherwise, the total number of non-zero values in the array is returned. int or cupy.ndarray of int
cupy.nonzero(a)[source]

Return the indices of the elements that are non-zero.

Returns a tuple of arrays, one for each dimension of a, containing the indices of the non-zero elements in that dimension.

Parameters: a (cupy.ndarray) – array Indices of elements that are non-zero. tuple of arrays
cupy.flatnonzero(a)[source]

Return indices that are non-zero in the flattened version of a.

This is equivalent to a.ravel().nonzero()[0].

Parameters: a (cupy.ndarray) – input array Output array, containing the indices of the elements of a.ravel() that are non-zero. cupy.ndarray
cupy.where(*args, **kwargs)[source]