我有两个Numpy数组:a: [2,5,3,7,9,1]
和b: [1,2,3,4,5,6]
。我想对a进行排序,并b
以与a
do的索引相同的方式包含shift元素。在这种情况下,a
将变为,[1,2,3,5,7,9]
并且b
将变为[6,1,3,2,4,5]
。我知道如何使用np.sort
排序a
,但是如何重新排列b
呢?
使用NumPy的argsort
:
>>> b[a.argsort()]
array([6, 1, 3, 2, 4, 5])
没有NumPy的替代方法:
>>> a, b = zip(*sorted(zip(a, b)))
>>> a
(1, 2, 3, 5, 7, 9)
>>> b
(6, 1, 3, 2, 4, 5)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句