np.where()用法

np.where有两种用法

1、np.where(condition,x,y) 当where内有三个参数时，第一个参数表示条件，当条件成立时where方法返回x，当条件不成立时where返回y。
2、np.where(condition) 当where内只有一个参数时，那个参数表示条件，当条件成立时，where返回的是每个符合condition条件元素的坐标,返回的是以元组的形式。

代码示例：

#用法一
#当self.net_input(X)返回的值大于等于0.0时，where返回1，否则返回0
np.where(self.net_input(X) >= 0.0, 1, 0)

#用法二
import numpy as np

a = np.array([2, 4, 6, 8, 10])
# 只有一个参数表示条件的时候
print(np.where(a > 5))

# out
(array([2, 3, 4], dtype=int64),)

参考：https://blog.csdn.net/island1995/article/details/90200151