np.where()简要介绍

函数

np.where(condition,x,y)

函数作用

1.满足条件(condition)，输出x，不满足输出y

2.若参数只有condition，而没有x和y，则输出非0元素的坐标

condition 条件，如input>8 x 满足condition输出的值 y 不满足condition输出的值

示例

import numpy as np 
    
a = np.array([2,4,6,8,10])
print(np.where(a>5,1,0))
print(np.where(a>5))

>>>[0 0 1 1 1]

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