2010-08-16 2 views
1

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

спасибо так много,

Diederik

+1

Может быть, я с видом что-то, но не будет достаточно простого сортировки и следующего разреза? – Wolph

+0

yep, не думал об этом (я знал, что я что-то пропускаю :)), если вы разместите его в качестве ответа, тогда я дам вам кредиты. – DrDee

ответ

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:] 
+0

Не должно быть 'np.sort (some_array) [:, - k:]'? –

Смежные вопросы