python之Numpy 将一个数组的所有非零元素组合到另一个数组中

在 Numpy 中，我有两个表示图像的三维数组。我试图在第一个图像上创建第二个图像的叠加层，所以我想用第二个数组中的相应元素替换第一个数组中的所有元素，但只有当它们不为零时。有什么简单的方法可以做到这一点吗？

请您参考如下方法:

这似乎是 np.where 的完美用例 ...

new_arr = np.where(second == 0, first, second) 

我已经在不合适的地方进行了替换(创建一个新数组而不是修改原始数组)，但这通常没问题...