SciPy

numpy.ma.masked_array.argpartition

masked_array.argpartition(kth, axis=-1, kind='introselect', order=None)[source]