Python获取numpy数组中最大的5个元素（保持原顺序）

本文主要演示numpy的argsort()函数的用法。这个函数的返回值是数组中的元素排序后的原下标，例如np.argsort([3,1,2])的返回结果是array([1, 2, 0], dtype=int64)，表达的是意思是原来下标1对应的元素最小，然后是原来下标2的元素，最后是原来下标0的元素最大。

下面的小代码演示了该函数的用法，并在最后按数组中原来的位置顺序返回了最大的5个元素。

>>> import numpy as np

>>> x = np.random.randint(1, 100, 10) # 随机整数

>>> x

array([84, 34, 22, 67, 5, 6, 76, 36, 99, 30])

>>> np.argsort(x) # 排序后的原下标

array([4, 5, 2, 9, 1, 7, 3, 6, 0, 8], dtype=int64)

>>> x[np.argsort(x)] # 按升序访问元素返回新数组

array([ 5, 6, 22, 30, 34, 36, 67, 76, 84, 99])

>>> x[sorted(np.argsort(x)[-5:])] # 按原来的顺序返回最大的5个数

array([84, 67, 76, 36, 99])