根據條件將 numpy 子數組添加到 numpy 數組

Question

我有一個 2d numpy 數組和一個 2d numpy 子數組，我想根據條件將其添加到原始數組。 我知道您可以像這樣將 2d 子數組添加到數組中：

original_array[start_x:end_x, start_y:end_y] = sub_array 

但我不知道如何有效地只添加大於 0 的 sub_array 值？

例子：

orginal_array = np.array([2,2],[2,2],[2,2],[2,2])
sub_array = np.array([0,0],[1,1],[0,1],[0,0])
expected_result = np.array([2,2], [1,1], [2,1], [2,2])

Answer 1

您可以根據條件>,< 0進行索引並添加 arrays。

orginal_array * (sub_array <= 0) + sub_array * (sub_array > 0)

array([[2, 2],
       [1, 1],
       [2, 1],
       [2, 2]])

Answer 2

另一種方法是使用np.where function 作為：

np.where(sub_array > 0, sub_array, original_array)

Output：

array([[2, 2], 
       [1, 1], 
       [2, 1], 
       [2, 2]])

Answer 3

嘗試，

sub_array2 = np.select([sub_array>0],[sub_array])
original_array[start_x:end_x, start_y:end_y] = sub_array2