如何使用另一个Numpy数组设置多维Numpy数组的单个元素？

Question

如果我们有一个类似于numpy数组的对象：

Array = np.zeros((2, 10, 10))

我们希望设置其中的一个元素，该元素由另一个元素给出

indexes = np.array([0,0,0])

我们应该如何做到这一点呢？

Array[indexes] = 5 

将Array的第一个维度的每个元素设置为5

- Fosa

---

你也可以使用列表（而不是numpy.array）：Array [0, 0, 0] = 5 只将第一个零设置为5。 - Lucas

Answer 1

使用a作为数据数组，idx作为索引数组，每行对应于要在数据数组中设置的一个元素，则可以执行以下操作-

a[tuple(idx.T)] = 5

样例运行 -

In [94]: a = np.zeros((2,2,3),dtype=int)

In [95]: idx = np.array([[0,0,0],[1,1,0],[0,1,2]])

In [96]: a[tuple(idx.T)] = 5

In [97]: a
Out[97]: 
array([[[5, 0, 0],
        [0, 0, 5]],

       [[0, 0, 0],
        [5, 0, 0]]])

In [98]: a[tuple(idx.T)] = [5,10,15] # or set different values

In [99]: a
Out[99]: 
array([[[ 5,  0,  0],
        [ 0,  0, 15]],

       [[ 0,  0,  0],
        [10,  0,  0]]])

或者，我们可以使用np.ravel_multi_index计算线性索引，然后使用np.put进行赋值，像这样 -

np.put(a,np.ravel_multi_index(idx.T,a.shape),5)

如果您要处理三维数组，我们可以对三维索引进行切片并赋值给另一种方法，例如 -

a[idx[:,0],idx[:,1],idx[:,2]] = 5

如果只需要设置一个元素，只需执行以下操作 -

a[tuple(idx)] = 5

示例运行 -

In [118]: a = np.zeros((2,2,3),dtype=int)

In [119]: idx = np.array([0,0,0])

In [120]: a[tuple(idx)] = 5

In [121]: a
Out[121]: 
array([[[5, 0, 0],
        [0, 0, 0]],

       [[0, 0, 0],
        [0, 0, 0]]])

- Divakar