两种用法
np.where(condition,X,Y):满足条件输出X,不满足输出Y
这点的话再数据挖掘的规约中显得非常好用
>>> a = [2,3,4,5,8,7,9,7]
>>> a = np.array(a)
>>> np.where(a > 5 ,1,-1)
array([-1, -1, -1, -1, 1, 1, 1, 1])
np.where(condition):返回满足条件的数的索引
如果为二维数组返回的是行列索引,先行后列,有点像np.nonzero()的返回值
>>> indices = np.where(a <5)
>>> indices
(array([0, 1, 2], dtype=int64),)
>>> b = a[np.where(a <5)]
>>> b
array([2, 3, 4])