np.where怎么使用

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

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

其中，condition 是一个条件表达式，x 和 y 是可选参数，表示满足条件和不满足条件时的返回值。如果只有一个参数，则返回满足条件的元素的索引。

下面是一个简单的例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.where(a > 2)
print(b)

输出：

(array([2, 3, 4]),)

这表示在数组 a 中，大于 2 的元素的索引分别为 2、3、4。