向量化 numpy 掩码设置

Question

我有 3 个尺寸为

的 numpy 数组

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

如何矢量化以下内容？

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

Answer 1

只需使用integer-indexing -

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