使用另一个Numpy 2D数组的相应元素替换Numpy 2D数组中的元素,具体取决于条件

时间:2015-11-15 00:35:44

标签: python arrays numpy

我是Numpy的新手,我想知道是否有一种快速的方法来替换2D阵列中的元素(让我们称之为#34; A")满足特定条件与另一个元素的对应元素2D数组(让我们称之为" B"),同时将剩余元素的值保存在数组" A"没有达到这个条件;我应该提到" B"具有与" A"相同的形状。

提前多多感谢

1 个答案:

答案 0 :(得分:3)

说条件是element < 2。然后我们可以创建一个掩码,指出哪些单元符合条件:

mask = A < 2

并使用高级索引选择B的相应元素,并将其值分配给A的相应单元格:

A[mask] = B[mask]