如何基于索引数组重新排列数组

Question

我正在寻找一个单线解决方案，可以帮助我做以下工作。

如果我有

array = np.array([10, 20, 30, 40, 50])

我想根据输入顺序重新排列它。如果有一个名为arrange的numpy函数，它将执行以下操作：

newarray = np.arrange(array, [1, 0, 3, 4, 2])
print newarray

    [20, 10, 40, 50, 30]

形式上，如果要重新排序的数组是m x n，“index”数组是1xn，则顺序将由名为“index”的数组确定。

numpy有这样的功能吗？

Answer 1

如果要按降序排序：

a = np.array([1,2,3,4,5])
np.argsort(a)
> array([0, 1, 2, 3, 4])
np.argsort(-a)
> array([4, 3, 2, 1, 0])