根据条件将Numpy 2D数组中的元素替换为另一个Numpy 2D数组的相应元素

Question

我是新来的Numpy，我想知道是否有一种快速的方法来替换2D数组中满足特定条件的元素（让我们称之为“a”）与另一个2D数组的对应元素（我们称之为“B”），同时保留数组“a”中不满足此条件的其余元素的值；我要说的是“B”和“A”的形状相同。在

事先非常感谢

Answer 1

假设条件是element < 2。然后我们可以创建一个掩码，指示哪些细胞符合条件：

mask = A < 2

并使用高级索引选择B的相应元素，并将它们的值赋给A的相应单元格：

^{pr2}$