Searching Functions

Array API specification for functions for searching arrays.

A conforming implementation of the array API standard must provide and support the following functions.

Objects in API

argmax(x, /, *, axis=None, keepdims=False)

Returns the indices of the maximum values along a specified axis.

argmin(x, /, *, axis=None, keepdims=False)

Returns the indices of the minimum values along a specified axis.

count_nonzero(x, /, *, axis=None, keepdims=False)

Counts the number of array elements which are non-zero.

nonzero(x, /)

Returns the indices of the array elements which are non-zero.

searchsorted(x1, x2, /, *, side='left', sorter=None)

Finds the indices into x1 such that, if the corresponding elements in x2 were inserted before the indices, the order of x1, when sorted in ascending order, would be preserved.

where(condition, x1, x2, /)

Returns elements chosen from x1 or x2 depending on condition.