np.where函数是什么怎么用

np.where函数是numpy中的一个函数，它用于根据给定条件返回输入数组中的元素的索引。

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

参数说明：

condition：条件数组，即需要判断的条件。

x，y：可选参数，用于指定满足条件时输出的值，其中x为满足条件时输出的值，y为不满足条件时输出的值。如果不指定x和y，则返回满足条件的元素的索引。

使用示例：

import numpy as np

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

# 返回满足条件的元素的索引
print(np.where(arr > 2))

# 返回满足条件的元素的值
print(np.where(arr > 2, arr, 0))

输出结果为：

(array([2, 3, 4], dtype=int64),)
[0 0 3 4 5]

其中，第一个示例返回的是满足条件的元素的索引；第二个示例返回的是满足条件的元素的值，不满足条件则返回0。