Python Numpy 2D数组索引

Question

我对python和numpy很陌生。请问有人可以帮助我了解如何对用作索引的某些数组进行索引。我有以下六个2D阵列-

array([[2, 0],
   [3, 0],
   [3, 1],
   [5, 0],
   [5, 1],
   [5, 2]])

我想将这些数组用作索引，并将值10放入新的空矩阵的相应索引中。输出应如下所示：

array([[ 0,  0,  0],
   [ 0,  0,  0],
   [10,  0,  0],
   [10, 10,  0],
   [ 0,  0,  0],
   [10, 10, 10]])

到目前为止，我已经尝试过

    from numpy import*
    a = array([[2,0],[3,0],[3,1],[5,0],[5,1],[5,2]])
    b = zeros((6,3),dtype ='int32')
    b[a] = 10

但这给了我错误的输出。任何帮助请。

Answer 1

In [1]: import numpy as np
In [2]: a = np.array([[2,0],[3,0],[3,1],[5,0],[5,1],[5,2]])
In [3]: b = np.zeros((6,3), dtype='int32')

In [4]: b[a[:,0], a[:,1]] = 10

In [5]: b
Out[5]: 
array([[ 0,  0,  0],
       [ 0,  0,  0],
       [10,  0,  0],
       [10, 10,  0],
       [ 0,  0,  0],
       [10, 10, 10]])

工作原理：

如果您在工作分配中b使用 两个 numpy数组建立索引，

b[x, y] = z

然后认为NumPy的作为过的各元件同时移动x和中的每个元素y和中的每个元素z（我们称它们xval，yval和zval），并分配至b
[XVAL，利用yval]的值zval。当z为常数时，“z每次移动仅返回相同的值。

这就是我们想要的，与x作为第一列a和y是第二列a。因此，选择x = a[:, 0]和y = a[:, 1]。

b[a[:,0], a[:,1]] = 10

为什么b[a] = 10不起作用

当你写b[a]，认为NumPy的，如通过的每一个元素移动在创造一个新的数组a，（我们称之为每一个idx）和新的阵列中放置的价值b[idx]在的位置idx中a。

idx是中的值a。所以它是一个int32。 b形状为（6,3），b[idx]一行b形状为（3，）。例如，何时idx是

In [37]: a[1,1]
Out[37]: 0

b[a[1,1]] 是

In [38]: b[a[1,1]]
Out[38]: array([0, 0, 0])

所以

In [33]: b[a].shape
Out[33]: (6, 2, 3)

因此，让我们再次重申：NumPy的是通过创建的每个元素移动在新的阵列a和新阵列中放置的价值b[idx]在的位置idx中a。随着idx移动a，将创建形状为（6,2）的数组。但是由于b[idx]形状本身为（3，），因此在（6,2）形数组中的每个位置都放置了（3，）形值。结果是形状（6,2,3）的数组。

现在，当您进行类似

b[a] = 10

b[a]创建具有值的形状（6,2,3）的临时数组，然后执行分配。由于10是常数，因此此赋值将值10放置在（6,2,3）形数组中的每个位置。然后，将临时数组中的值重新分配给b。请参阅参考文档。因此，将（6,2,3）形数组中的值复制回（6,3）形b数组中。值彼此覆盖。但要点是您没有获得所需的任务。