np.where()函数的用法

np.where()函数的用法
1.根据判断条件返回下标

d = np.array([[4,-6,7],[4,-5,6]])
e =  np.where(d<0)
#return (array([0, 1]), array([1, 1]))

2.根据判断条件返回制定值

a = 10
b = np.where(a > 0, 1.0, 0.0)
print(b)
#retuen 1.0
``