SciPy

numpy.ma.masked_array.argpartition

masked_array.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