我正在尝试使用另一个数组的索引从一个数组中获取值。我实现了下面的代码,该代码可以正常工作,但尚未向量化。我想知道是否有一个更简单的矢量化版本。
import numpy as np
a = np.random.randint(0, 10, (4, 3))
b = np.random.randint(0, 3, (4, 8), dtype=np.int)
c = np.zeros((4, 8), dtype=np.int)
for i in range(4):
c[i] = a[i][b[i]]
我尝试过[b],但这给了我一个错误。