向量化NumPy的屏蔽設置

Question

我有尺寸

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

我怎樣才能向量化以下3 numpy的陣列？

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

Answer 1

只需使用integer-indexing -

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