gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 07:37:17 26 4
gpt4 key购买 nike

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

最佳答案

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

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

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

关于python - Numpy 将一个数组的所有非零元素组合到另一个数组中,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31862704/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com