学习np.mask

import numpy as np

'''
学习np.mask
针对特定的遥感数据（某一行要是有掩码，则全行为掩码，可以全部删掉，所以只需判断每行的第一个数是否为掩码）
'''

a = np.array([[1,2,3],
              [4,5,6],
              [2,2,2],
              [3,3,3]])
w,h = a.shape
mask_a = (a>2)
aa = np.ma.masked_array(a,mask=mask_a)
#aa.mask = np.ma.nomask

print(aa)
print("---------------------------")
aaa = []
i = 0
for i in range(w):
    #print(aa[i][0])
    if aa[i][0] is np.ma.masked:
        pass
    else:
        aaa.append(aa[i])

print(np.array(aaa))