np.where()是干什么的

np.where()是NumPy库中的一个函数，用于根据一个条件从数组中选取元素。它的语法如下：

np.where(condition[, x, y])

其中，condition表示条件，可以是一个布尔数组或一个条件表达式；x和y是可选参数，表示当条件为True时返回x的相应元素，否则返回y的相应元素。

np.where()的返回值是一个数组，它的元素是从x和y中根据条件选择的元素。如果省略了x和y，则返回一个包含所有满足条件的元素的下标的元组。