在 Numpy 中,我有两个表示图像的三维数组。我试图在第一个图像上创建第二个图像的叠加层,所以我想用第二个数组中的相应元素替换第一个数组中的所有元素,但只有当它们不为零时。有什么简单的方法可以做到这一点吗?
最佳答案
这似乎是 np.where
的完美用例 ...
new_arr = np.where(second == 0, first, second)
我已经在不合适的地方进行了替换(创建一个新数组而不是修改原始数组),但这通常没问题...
关于python - Numpy 将一个数组的所有非零元素组合到另一个数组中,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31862704/