如果列表按降序排序,我需要返回列表元素将具有的索引。
例如给定 np.array([2,3,1,4,5])
我想接收 array([3,2,4,1,0] )
。
有没有简单的方法来获取这样的数组?
最佳答案
您想要的是元素按降序排列的排名。
a = np.array([2,3,1,4,5])
sorted_indices = np.argsort(-a)
ranks = np.empty_like(sorted_indices)
ranks[sorted_indices] = np.arange(len(a))
结果:
>>> ranks
array([3, 2, 4, 1, 0])
关于python - 按降序排列 numpy 列表的元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66275015/