python - 向量化 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