np.where()函数

`np.where()`是numpy库中的一个函数，它的作用是根据条件返回给定数组中符合条件的元素的下标或值。

`np.where()`函数的语法如下：

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

其中，参数说明如下：

- `condition`：条件，可以是bool型数组或bool型表达式。
- `x`：满足条件(condition)时返回的数组，可以是numpy数组或标量。
- `y`：不满足条件(condition)时返回的数组，可以是numpy数组或标量。如果没有指定，则返回满足条件 (condition) 的元素的索引。

`np.where()`函数返回一个元组，包含符合条件的元素的下标或值。

例如，对于一个numpy数组`arr`，我们可以使用`np.where()`函数来查找其中所有大于0的元素的下标：

import numpy as np

arr = np.array([-2, 0, 3, -1, 5, -4, 6])

indexes = np.where(arr > 0)

print(indexes)

输出结果为：

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

其中，`(array([2, 4, 6], dtype=int64),)`表示符合条件的元素在原数组中的下标为2、4、6。