python - 撤消或反向 argsort()，python

Question

给定一个数组 'a' 我想按列对数组进行排序 sort(a, axis=0) 对数组做一些事情，然后撤消排序。我的意思不是重新排序，而是基本上颠倒每个元素的移动方式。我假设 argsort() 是我需要的，但我不清楚如何使用 argsort() 的结果对数组进行排序或更重要的是应用反向/反向argsort()

这里有一点细节

我有一个数组a，shape(a) = rXc我需要对每一列进行排序

aargsort = a.argsort(axis=0)  # May use this later
aSort = a.sort(axis=0)

现在平均每一行

aSortRM = asort.mean(axis=1)

现在用行平均值替换行中的每个列。还有比这更好的方法吗

aWithMeans = ones_like(a)
for ind in range(r)  # r = number of rows
    aWithMeans[ind]* aSortRM[ind]

现在我需要撤消我在第一步中所做的排序。 ????

Answer 1

对于您实际尝试解决的问题，可能有比这更好的解决方案(执行 argsort 通常会排除实际排序的需要)，但是您可以这样做:

>>> import numpy as np
>>> a = np.random.randint(0,10,10)
>>> aa = np.argsort(a)
>>> aaa = np.argsort(aa)
>>> a # original
array([6, 4, 4, 6, 2, 5, 4, 0, 7, 4])
>>> a[aa] # sorted
array([0, 2, 4, 4, 4, 4, 5, 6, 6, 7])
>>> a[aa][aaa] # undone
array([6, 4, 4, 6, 2, 5, 4, 0, 7, 4])