npwhere的详细用法

np.where是numpy库中的一个函数，用于根据指定的条件返回符合条件的元素下标或者元素本身的值，具体用法如下：

1. 返回符合条件的元素下标

语法：np.where(condition)

参数：

condition：指定的条件，可以是一个bool类型的数组或者条件表达式。

返回值：返回一个元组，包含符合条件的元素下标。

示例代码：

import numpy as np

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

print(result)

输出：

(array([3, 4]),)

2. 返回符合条件的元素本身的值

语法：np.where(condition, x, y)

参数：

condition：指定的条件，可以是一个bool类型的数组或者条件表达式。

x：如果condition中的元素为True，则返回x对应的元素值。

y：如果condition中的元素为False，则返回y对应的元素值。

返回值：返回一个数组，包含符合条件的元素的值。

示例代码：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = np.where(arr > 3, arr, 0)

print(result)

输出：

[0 0 0 4 5]

以上就是np.where的详细用法。