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