Vectorisation numpy masque réglage

Question

J'ai 3 tableaux de numpy avec des dimensions

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

Comment puis-je vectoriser ce qui suit?

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

Answer 1

Utilisez simplement integer-indexing -

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