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.
Returns:Array of the same type and shape as a.
Return type: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.

See also

numpy.sort()

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.
Returns:

The indices of the maximum of a along an axis.

Return type:

cupy.ndarray

See also

numpy.argmax()

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.
Returns:

The indices of the minimum of a along an axis.

Return type:

cupy.ndarray

See also

numpy.argmin()

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
Returns:

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.

Return type:

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
Returns:Indices of elements that are non-zero.
Return type:tuple of arrays

See also

numpy.nonzero()

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
Returns:Output array, containing the indices of the elements of a.ravel() that are non-zero.
Return type:cupy.ndarray
cupy.where(*args, **kwargs)[source]