Numpy: как получить k max значений из каждой строки?

Question

Я хочу получить k max значений из каждой строки в массиве numpy. Я копался в документации, но не смог найти ответ (вероятно, я смотрю не в том месте). У кого-нибудь есть простой фрагмент кода, который делает это?

Большое спасибо,

Diederik

Answer 1

Вот простое решение:

some_array = numpy.random.randint(0, 10, 10)
top_items = numpy.sort(some_array)[-k:]

Или как лямбда:

max_values = lambda k, array: numpy.sort(array)[-k:]