python - 如何排序 unsort : array(1). array(2) -> array(3).unsort (reversed array(1).sort

Question

您如何对结果进行排序、操作和取消排序？

假设我有一个 float 组p1 = 0.15,0.3, 0.25, 0.12, ...。它被排序为:p2 = sort(p1)。函数(以 p2 作为输入进行操作)产生 p3:p3 = f(p2, x, y, ...) 对于某些函数 f。

如何以最聪明的方式对 p3 进行排序？ (与 p1 的排序相反)

即:p4 = unsort(p3) <- p4 未排序为与 p1 相同的顺序，用于比较(x-plot) p1?

Answer 1

你需要一个双参数排序来保持顺序:

In [6]: a
Out[6]: array([5, 4, 8, 3, 6, 1, 2, 4, 9, 6])

In [7]: b=sort(a)

In [8]: b
Out[8]: array([1, 2, 3, 4, 4, 5, 6, 6, 8, 9])

In [9]: ii=a.argsort().argsort()

In [10]: c=b*b

In [11]: c
Out[11]: array([ 1,  4,  9, 16, 16, 25, 36, 36, 64, 81])

In [12]: c[ii]
Out[12]: array([25, 16, 64,  9, 36,  1,  4, 16, 81, 36])