本文主要演示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])