Cherish's Notebook
切换到深色模式
菜单
Wiki--浏览次数--访问次数--跳出率--平均停留

numpy

15.比较掩码

比较操作,会返回与参与运算数组形状相同的数组,其中,满足条件的为True,不满足的为False

CODE
a=np.array([[1,2,3],[11,12,13],[21,22,23]])
a<20#结果是一个由True,False组成的数组
a[a<20]#小于20的元素会存到这个数组中
#结果:
#array([ 1,  2,  3, 11, 12, 13])