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循环,但我明确地想避免使用一个,因为我正在使用更高维数组执行此过程,结果是程序中的大幅减速。
答案 0 :(得分:3)
它只是a[b]
。 (额外的单词,因为StackOverflow需要30个字符。我会发布一个演示,但我不能从这里运行一个,我不想假装输出。)