python - 如何根据索引数组重新排列数组

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”数组是 1 x n，则排序将由名为“index”的数组确定。

numpy有这样的功能吗？

Answer 1

您可以直接使用“索引”列表，也可以使用索引数组:

>>> arr = np.array([10, 20, 30, 40, 50])
>>> idx = [1, 0, 3, 4, 2]
>>> arr[idx]
array([20, 10, 40, 50, 30])

如果 idx 已经是 ndarray 而不是 list，它往往会快得多，即使它可以以任何一种方式工作:

>>> %timeit arr[idx]
100000 loops, best of 3: 2.11 µs per loop
>>> ai = np.array(idx)
>>> %timeit arr[ai]
1000000 loops, best of 3: 296 ns per loop