矢量化numpy掩码设置

Question

我有3个具有维度的numpy数组

M (NxC)     # mask
y (N,)     
values (N,)

如何将以下内容矢量化

for i in range(N):
      M[i][y[i]] = values[i]

Answer 1

只需使用^{}-

M[np.arange(N),y] = values