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

Question

I am new to Numpy and I was wondering if there is a fast way to replace elements in a 2D array (lets call it "A") that are meeting a specific condition with their corresponding elements of another 2D array (lets call it "B"), and at the same time keep the values of the remaining elements in array "A" that didn't meet this condition; I should mention that "B" has the same shape as "A".

Thanks a lot in advance

Answer 1

Say the condition is element < 2 . Then we can create a mask indicating which cells match the condition:

mask = A < 2

and use advanced indexing to select the corresponding elements of B and assign their values to the corresponding cells of A :

A[mask] = B[mask]