【Numpy】np.where()的两种用法

两种用法

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])