numpy.ndarray.argpartition

метод

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

Возвращает индексы, которые бы разделили этот массив.

Обратитесь к numpy.argpartition для полной документации.

Новая в версии 1.8.0.

См. также