python - NumPy 数组的返回顺序

Question

我正在尝试返回一个数组，该数组具有数组中每个值的排名。例如，给定以下数组:

import numpy as np
arr1 = np.array([4, 5, 3, 1])

我想返回数组:

array([2, 3, 1, 0])

使得返回数组中的值表示数组的升序(即返回数组中的值表示哪个最大)。使用 argsort，我只能告诉值应该如何重新排序:

arr1.argsort()
array([3, 2, 0, 1])

如果不清楚请告诉我。

Answer 1

可能有更好的方法，但我总是这样做argsort().argsort():

>>> import numpy as np
>>> a = np.random.random(5)
>>> a
array([ 0.54254555,  0.4547267 ,  0.50008037,  0.20388227,  0.13725801])
>>> a.argsort().argsort()
array([4, 2, 3, 1, 0])