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)