numpy数组替换部分元素

我有两个numpy浮点型数组,a和b,不用循环的方式,将数组a中和数组b相同的元素替换为nan?

试试这个。

>>> import numpy as np
>>> a = np.array([3.4, 2.5, 1.8, 4.7, 5.6, 2.1])
>>> b = np.array([2.5, 4.7, 5.6])
>>> np.where(np.isin(a,b), np.nan, a)
array([3.4, nan, 1.8, nan, nan, 2.1])

 

数组是一样大小吗?数组大小不一样写法就不一样了。

您好,我是有问必答小助手,你的问题已经有小伙伴为您解答了问题,您看下是否解决了您的问题,可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案,可以点击【采纳】按钮,给回答的小伙伴一些鼓励哦~~

ps:问答VIP仅需29元,即可享受5次/月 有问必答服务,了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632

非常感谢您使用有问必答服务,为了后续更快速的帮您解决问题,现诚邀您参与有问必答体验反馈。您的建议将会运用到我们的产品优化中,希望能得到您的支持与协助!

速戳参与调研>>>https://t.csdnimg.cn/Kf0y