numpy 获取数组在升序中的序号

• • 要干嘛

为了便于理解：
假设有一组人的身高（一个数组），要知道每个人的身高在总体中（升序）的序号，例如[1.80,1.90,1.70],得到[2,3,1].

才疏学浅，只知道numpy的argsort是得到应用排序后，当前位置在原数组中的索引。即[1.80,1.90,1.70],得到[2,0,1].解释：因为升序后，第一个人的身高应为1.80，该人在原数组的索引为2，所以结果中第一个数为2.
对此，稍微改造一下，得到我们想要的结果。(没系统学过numpy，不知道有没有现成的方法)
以下是代码:

x = np.array([1.80, 1.90, 1.70, 2.0, 1.60, 1.66])
y = np.argsort(x)
r = np.ones(len(x), dtype=int)
for i in range(len(x)):
    #排序从1开始，所以加1
    r[y[i]] = i + 1
print("原先数组:", x)
print("应用升序后当前位置在原数组的索引:", y)
print("原数组中元素在升序中的序号:", r)

得到结果

如果要得到降序中的序号的话，稍微改造一下
y = np.argsort(x) >> y = np.argsort(-x)

x = np.array([1.80, 1.90, 1.70, 2.0, 1.60, 1.66])
y = np.argsort(-x)
r = np.ones(len(x), dtype=int)
for i in range(len(x)):
    #排序从1开始，所以加1
    r[y[i]] = i + 1
print("原先数组:", x)
print("应用降序后当前位置在原数组的索引:", y)
print("原数组中元素在降序中的序号:", r)

得到结果