Numpy用另一个数组的特定行和列替换一个数组的特定行和列

Question

我试图替换Numpy数组的特定行和列，如下所示。

数组a和b的值最初如下：

a = [[1 1 1 1]
     [1 1 1 1]
     [1 1 1 1]]

b = [[2 3 4 5]
     [6 7 8 9]
     [0 2 3 4]]

现在，根据一定的概率，我需要执行的的elementwise替换a与值b （比如，产生一个随机数，后r ，0和1之间的每个元素，我将取代的元件a与的b如果r> 0.8）。

我如何使用numpy / scipy在Python中以高性能执行此操作？

Answer 1

随着掩蔽 。 我们首先生成一个具有相同尺寸，随机数的矩阵，并检查它们是否大于0.8 ：

mask = np.random.random(a.shape) > 0.8

现在，我们可以分配的值b ，其中mask是True上对应的指数a ：

a[mask] = b[mask]

例如：

>>> a
array([[1., 1., 1., 1.],
       [1., 1., 1., 1.],
       [1., 1., 1., 1.]])
>>> b
array([[2, 3, 4, 5],
       [6, 7, 8, 9],
       [0, 2, 3, 4]])
>>> mask = np.random.random(a.shape) > 0.8
>>> mask
array([[ True, False, False, False],
       [ True, False, False, False],
       [False, False, False, False]])
>>> a[mask] = b[mask]
>>> a
array([[2., 1., 1., 1.],
       [6., 1., 1., 1.],
       [1., 1., 1., 1.]])

所以这里mask是True （因为0.8相当高，我们期望平均2.4这样的值），我们分配相应的b值。