ndarray 过滤条件

import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]]) #原始数据
e = (a > 6) |  (a <2)    #构造对原始数据进行筛选的条件
a4 = np.where(e,a,0) #把满足条件的选择出来,原封不动的保存,不满足条件的元素置零
                                   #本质上,就是把矩阵元素,按照条件分类.
a5 = a[e]                     #把满足条件的元素选择出来,构成新的元素子集合

多重筛选条件

tmp = np.where(((label_np != 0) & (label_np != label_ID)), 0, label_np)
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容