numpy funkcja ustawić elementy tablicy do wartości danej lista indeksów

Question

szukam funkcji numpy że zrobi to równowartość:

indices = set([1, 4, 5, 6, 7]) 
zero = numpy.zeros(10) 
for i in indices: 
    zero[i] = 42 

Answer 1

może po prostu dać mu listę indeksów:

indices = [1, 4, 5, 6, 7] 
zero = numpy.zeros(10) 
zero[indices] = 42 

Answer 2

Jeśli masz ndarry:

>>> x = np.zeros((3, 3, 3)) 
>>> y = [0, 9, 18] 
>>> x 
array([[[ 0., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]], 

     [[ 0., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]], 

     [[ 0., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]]]) 
>>> np.put(x, y, 1) 
>>> x 
array([[[ 1., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]], 

     [[ 1., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]], 

     [[ 1., 0., 0.], 
     [ 0., 0., 0.], 
     [ 0., 0., 0.]]])