np.where()函数

np.where()函数是 numpy 库中的一个函数，用于根据一个给定的条件，返回满足条件的数组元素的索引或值。它的语法如下：

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

其中，参数说明如下：

- `condition`：指定的条件，可以是一个布尔数组或表达式；
- `x`：可选参数，如果满足条件，则返回值为 `x`，否则返回满足条件的元素的索引；
- `y`：可选参数，如果满足条件，则返回值为 `y`，否则返回满足条件的元素的索引。

`np.where()`函数的返回值是一个元组，其中第一个元素是满足条件的元素的索引。如果只有一个参数，则返回满足条件的元素的索引；如果有两个参数，则返回满足条件的元素的值。如果有三个参数，则返回满足条件的元素的值，否则返回满足条件的元素的索引。