我是Numpy的新手,我想知道是否有一种快速的方法来替换2D阵列中的元素(让我们称之为#34; A")满足特定条件与另一个元素的对应元素2D数组(让我们称之为" B"),同时将剩余元素的值保存在数组" A"没有达到这个条件;我应该提到" B"具有与" A"相同的形状。
提前多多感谢
答案 0 :(得分:3)
说条件是element < 2
。然后我们可以创建一个掩码,指出哪些单元符合条件:
mask = A < 2
并使用高级索引选择B
的相应元素,并将其值分配给A
的相应单元格:
A[mask] = B[mask]