python中np.where()

np.where()函数是numpy中的一个函数，用于根据指定条件返回相应的元素的索引或值。其语法如下：

np.where(condition, x, y)

其中，condition是一个布尔型数组，表示要检查的条件；x和y是两个数组，表示条件为True时返回的值和条件为False时返回的值。如果只传入一个参数，则返回条件为True的元素的索引。

例如，下面的代码演示了如何使用np.where()函数来找到一个数组中大于0的元素的索引：

import numpy as np

arr = np.array([-2, 0, 3, -5, 6])
indices = np.where(arr > 0)
print(indices)

输出结果为：

(array([2, 4]),)

这表示arr数组中第2个和第4个元素大于0。注意，np.where()返回的是一个元组，其中第一个元素是一个数组，表示符合条件的元素的索引。如果要获取这些元素的值，可以使用arr[indices]。