chức năng NumPy để thiết lập các yếu tố của mảng đến một giá trị được đưa ra một danh sách các chỉ số

Question

Tôi đang tìm kiếm một chức năng NumPy rằng sẽ làm tương đương với:

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

Answer 1

Bạn chỉ có thể cung cấp cho nó một danh sách các chỉ số:

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

Answer 2

Nếu bạn có một 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.]]])