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)
得到结果