SciPy

numpy.ndarray.argpartition

ndarray.argpartition(kth, axis=-1, kind='quickselect', order=None)

Returns the indices that would partition this array.

Refer to numpy.argpartition for full documentation.

New in version 1.8.0.

See also

numpy.argpartition
equivalent function