Нахождение K наибольшее в nparray (значение и аргумент) - PullRequest
0 голосов
/ 01 февраля 2019

Я хотел бы найти N самых больших элементов и их аргументы в массиве numpy.

Я использую найденные разделы, но не могу получить 5 самых больших элементов в массиве numpy.Вот пример:

a =[ 0.409,  0.355, -0.403,  0.307, -0.826, -0.66,  -1.582, -0.112, -0.244, -0.954]    
z = bn.argpartition(-a, 5)[:5]         
z = apodosi.argsort()[-5:]         
z = heapq.nlargest(5, a)   

Я возвращаю массив некоторое время, и я хотел бы получить 0,1,3,5,7 в качестве аргументов и, если возможно, их значения.Я все еще новичок в программировании!

1 Ответ

0 голосов
/ 01 февраля 2019

Самый простой способ, вероятно, таков:

array = [5,3,1,2,4,7,9,8,6]
array.sort(reverse=True) # [9,8,7,...,2,1]
greatest = array[0:5] # [9,8,7,6,5]

Я полагаю, что numpy массивы будут работать так же.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...