NumPy：取消排序/撤消反向/降序排序

我可以对 numpy 数组进行就地反向排序（降序排序），但我还需要能够稍后对其进行取消排序（撤消）。

给出一个未排序的例子：

a = np.array([-1, -2,  1, -3,  2,  0])

I tried:

i = a[::-1].argsort().argsort()  # BAD attempt to store original index
# i = array([3, 5, 0, 4, 1, 2])

a[::-1].sort()  # in-place reverse sort (works correctly)
# a= array([ 2,  1,  0, -1, -2, -3])

a = a[i]  # FAILS to restore original a
# a = array([-1, -3,  2, -2,  1,  0])

上面的方法不起作用。什么是正确的i那行得通吗？假设数组非常大，因此我们不想制作任何不必要的副本。

下面的习惯用法比第二个 argsort (O(n log n)) 更便宜 (O(n))。

示例数组x与排序顺序y

>>> x = np.random.random(10)
>>> y = x.argsort()[::-1]

构建逆i

>>> i = np.empty_like(y)
>>> i[y] = np.arange(y.size)

Check:

>>> x
array([0.44257134, 0.573158  , 0.07762422, 0.31507426, 0.43414726,
       0.34923861, 0.22161337, 0.14090133, 0.66903264, 0.38888105])
>>> x[y]
array([0.66903264, 0.573158  , 0.44257134, 0.43414726, 0.38888105,
       0.34923861, 0.31507426, 0.22161337, 0.14090133, 0.07762422])
>>> x[y][i]
array([0.44257134, 0.573158  , 0.07762422, 0.31507426, 0.43414726,
       0.34923861, 0.22161337, 0.14090133, 0.66903264, 0.38888105])