python里的numpy的array有个小技巧，可以方便的处理数据

import numpy as np
a=np.array([1,-1,2])
b=a>0
c=a[b]  #a[a>0]
print(a,b,c)

输出为：

[ 1 -1  2] [ True False  True] [1 2]

这样就可以直接对数组a做大小判断，直接获得符合条件的新数组。