Numpy:如何基于索引数组的重新排列数组?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (45)

我正在寻找一个单行解决方案,它将帮助我完成以下工作。

如果我有

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

我想根据输入顺序重新排列。如果有一个numpy函数arrange,它将做以下工作:

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

    [20, 10, 40, 50, 30]

形式上,如果要重新排序的数组为mxn,而“index”数组为1xn,则排序将由称为“index”的数组确定。

Numpy有这样的功能吗?

提问于
用户回答回答于

您可以直接使用“索引”列表,作为索引数组:

>>> 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

扫码关注云+社区