考虑以下几点:
In [51]: arr = np.arange(6, 10)
In [52]: idx = np.random.randint(4, size=(3, 4))
In [53]: idx
Out[53]:
array([[0, 3, 3, 1],
[1, 3, 3, 2],
[1, 1, 1, 1]])
In [54]: result = np.empty_like(idx)
In [55]: for i in range(idx.shape[0]):
...: result[i] = arr[idx[i]]
...:
In [56]: result
Out[56]:
array([[6, 9, 9, 7],
[7, 9, 9, 8],
[7, 7, 7, 7]])
如何对for
循环进行矢量化?我找不到通过索引矩阵“多次”访问一维数组的方法,其中每一行都是索引数组
如注释中所述,您可以使用
idx
数组简单地索引到数组arr
如果你想要一个不那么神奇,更具启发性的方法,那么下面的方法会更有帮助
相关问题 更多 >
编程相关推荐