使用其他数组索引数组(不使用for循环)

时间:2015-11-18 21:58:29

标签: python arrays numpy indexing

numpy中是否有一种方法可以使用另一个数组索引数组。例如,如果我有这个数组:

a = [1.0, 2.0, 3.0, 4.0]

和这个数组

b = [3, 1, 1, 2]

是否有一种索引使用b的方法,使得新数组的第0个组件是a的第3个组件,新数组的第1个组件是a的第1个组件,依此类推。

换句话说就是这个

[4.0, 2.0, 2.0, 3.0]

我知道这很容易用for循环,但我明确地想避免使用一个,因为我正在使用更高维数组执行此过程,结果是程序中的大幅减速。

1 个答案:

答案 0 :(得分:3)

它只是a[b]。 (额外的单词,因为StackOverflow需要30个字符。我会发布一个演示,但我不能从这里运行一个,我不想假装输出。)