矢量化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