问如何在numpy中进行特殊的选择性掩蔽
EN

Stack Overflow用户

提问于 2020-10-20 11:57:21

回答 1查看 39关注 0票数 0

我有两个2d NumPy数组。我需要屏蔽数组1上的数组2，这样，array2中的任何非零值都应该在array1中被替换，但其他值应该保持不变。我需要函数选择掩码，在那里我可以输入这两个2d数组，并返回输出。例：

array1=np.array([[1,2,3],[0,0,0],[3,3,3]])
array2=n.array([[0,0,0],[0,0,0],[7,8,9]])
result_array=selective_mask(array1,array2)

结果数组应该是[[1,2,3],[0,0,0],[7,8,9]]，因为您只能看到元素7、8、9被切换，因为这只是array2中的非零元素。

回答已采纳

发布于 2020-10-20 12:00:11

我想出了一个解决办法。我用了np.where

np.where(array2!=0,array2,array1)

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/64444678

复制

相似问题

问如何在numpy中进行特殊的选择性掩蔽EN