没有for循环的数组赋值[重复]

Question

我在为 numpy 结构赋值时遇到问题。例如，假设我有一个 numpy 数组 Src(N*K) 和一个索引数组 Index(N*1)，其值在 0~K-1 之间。我的任务是根据索引将Src 的值分配给数组Target(N*1)。

一种天真的方法是使用 for 循环：

for i in xrange(Index.shape[0]):
    Target[i,:] = Src[i,Index[i]]

但是，我相信应该有其他优雅的方式来实现此任务而无需 for 循环。那么有人可以帮助我吗？

Answer 1

你可以这样做：

>>> Target = np.zeros((4, 4))
>>> Src = np.ones((4, 4))
>>> Index = np.array([1, 2])
>>> Target[i, :] = Src[i, Index, None]
>>> Target
array([[ 1.,  1.,  1.,  1.],
       [ 1.,  1.,  1.,  1.],
       [ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.]])