如果满足条件则替换Numpy元素
我有一个需要处理的大型numpy数组，以便在满足条件的情况下将每个元素更改为1或0（稍后将用作像素蒙版）。数组中大约有800万个元素，而我目前的方法对于简化流程花费的时间太长：
for (y,x), value in numpy.ndenumerate(mask_data):
if mask_data[y,x]<3: #Good Pixel
mask_data[y,x]=1
elif mask_data[y,x]>3: #Bad Pixel
mask_data[y,x]=0
是否有一个numpy函数可以加快速度？ 问题来源于stack overflow
写回答
-
import numpy as np a = np.random.randint(0, 5, size=(5, 4)) a array([[4, 2, 1, 1], [3, 0, 1, 2], [2, 0, 1, 1], [4, 0, 2, 3], [0, 0, 0, 2]]) b = a < 3 b array([[False, True, True, True], [False, True, True, True], [ True, True, True, True], [False, True, True, False], [ True, True, True, True]], dtype=bool)
c = b.astype(int) c array([[0, 1, 1, 1], [0, 1, 1, 1], [1, 1, 1, 1], [0, 1, 1, 0], [1, 1, 1, 1]]) 您可以使用以下方法来缩短它：
c = (a < 3).astype(int)
2020-02-08 20:24:56赞同 展开评论 打赏
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
