如何“压缩排序”并行的numpy数组？

如何“压缩排序”并行的numpy数组？

b[a.argsort()]

应该可以。

运作方式如下。首先，您需要找到排序为a的排列。

argsort

是一种计算方法：

>>> a = numpy.array([2, 3, 1])>>> p = a.argsort()>>> p[2, 0, 1]

您可以轻松地检查这是否正确：

>>> a[p]array([1, 2, 3])

现在对b应用相同的排列。

>>> b = numpy.array([4, 6, 7])>>> b[p]array([7, 4, 6])