python – Numpy“类似”功能

假设我有两个数组,v和w:

v=np.arange(10)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
w=np.random.random(10,)
array([ 0.20224634,  0.19145386,  0.44607429,  0.53601637,  0.29148741,
        0.62670435,  0.95371219,  0.63634805,  0.48733178,  0.17155278])

我可以这样排序:

np.sort(w)
array([ 0.17155278,  0.19145386,  0.20224634,  0.29148741,  0.44607429,
        0.48733178,  0.53601637,  0.62670435,  0.63634805,  0.95371219])

我想以与w相同的方式排序v.例如.以便元素9移动到元素0,依此类推,直到v变为:

array([9, 1, 0, 4, 2, 8, 3, 5, 7, 6])

有没有一种简单的方法可以做到这一点,我错过了?

如果没有,你会怎么做?

最佳答案 您可以使用np.argsort获取订单:

order = np.argsort(w)

然后只对两个数组进行排序:

w = w[order]
v = v[order]
点赞