Python numpy 数组替换

Question

我有一个数组，例如:

array([[ 1,  1,  2,  0,  4],
       [ 5,  6,  7,  8,  9],
       [10,  0,  0, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22,  0, 24],
       [25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34],
       [35, 36, 37, 38, 39],
       [40, 41, 42, 43, 44],
       [45, 46, 47, 48, 49]])

我还有另外两个数组:

array([[   0,    0,    0,    0],
       [   0,    0,    0,    0],
       [   0, 2891,    0,    0],
       [   0,    0,    0,    0],
       [   0,    0,    0, 2891]])

和

array([[  0,   0,   0, 643],
       [  0,   0,   0,   0],
       [  0,   0, 643,   0],
       [  0,   0,   0,   0],
       [  0,   0,   0,   0]])

我想要的是从第二个数组到相应位置的第一个数组中选择值 2891，并从第三个数组到相应位置的第一个数组中选择值 643，这样最终的数组应该如下所示:

array([[   1,    1,    2,  643,    4],
       [   5,    6,    7,    8,    9],
       [  10, 2891,  643,   13,   14],
       [  15,   16,   17,   18,   19],
       [  20,   21,   22, 2891,   24],
       [  25,   26,   27,   28,   29],
       [  30,   31,   32,   33,   34],
       [  35,   36,   37,   38,   39],
       [  40,   41,   42,   43,   44],
       [  45,   46,   47,   48,   49]])

到目前为止我已经尝试过这个命令:

np.place(a,a<1, np.amax(b))

其中 a 指的是第一个数组，b 指的是第二个数组。它的作用只是用 2891 值替换所有 0 值。有人可以帮忙吗？

Answer 1

您可以使用 nonzero 方法找到 y 和 z 不为零的索引:

In [9]: y.nonzero()
Out[9]: (array([2, 4]), array([1, 3]))

In [10]: z.nonzero()
Out[10]: (array([0, 2]), array([3, 2]))

您可以通过fancing indexing 选择关联值:

In [11]: y[y.nonzero()]
Out[11]: array([2891, 2891])

并且您可以将这些值分配给 x 中的位置

In [13]: x[y.nonzero()] = y[y.nonzero()]

---

import numpy as np
x = np.array([[ 1,  1,  2,  0,  4],
              [ 5,  6,  7,  8,  9],
              [10,  0,  0, 13, 14],
              [15, 16, 17, 18, 19],
              [20, 21, 22,  0, 24],
              [25, 26, 27, 28, 29],
              [30, 31, 32, 33, 34],
              [35, 36, 37, 38, 39],
              [40, 41, 42, 43, 44],
              [45, 46, 47, 48, 49]])

y = np.array([[   0,    0,    0,    0],
              [   0,    0,    0,    0],
              [   0, 2891,    0,    0],
              [   0,    0,    0,    0],
              [   0,    0,    0, 2891]])

z = np.array([[  0,   0,   0, 643],
              [  0,   0,   0,   0],
              [  0,   0, 643,   0],
              [  0,   0,   0,   0],
              [  0,   0,   0,   0]])

x[y.nonzero()] = y[y.nonzero()]
x[z.nonzero()] = z[z.nonzero()]
print(x)

产量

[[   1    1    2  643    4]
 [   5    6    7    8    9]
 [  10 2891  643   13   14]
 [  15   16   17   18   19]
 [  20   21   22 2891   24]
 [  25   26   27   28   29]
 [  30   31   32   33   34]
 [  35   36   37   38   39]
 [  40   41   42   43   44]
 [  45   46   47   48   49]]