numpy中的np.where()函数

np.where(条件)，若条件满足，返回索引值
np.where(条件，x，y)，若条件满足，返回x，否则返回y