np.where怎么用

np.where() 是 numpy 库中的一个函数，用于根据一个条件从两个数组当中选择元素。

函数的使用格式如下：

np.where(condition, x, y)

参数说明：

- condition：一个布尔型数组或条件表达式。
- x：满足条件时的结果。
- y：不满足条件时的结果。

函数的返回值是一个数组，其元素来源于 x 和 y。

下面是一个示例代码：

import numpy as np

# 创建两个数组
a = np.array([0, 1, 2, 3, 4, 5])
b = np.array([10, 20, 30, 40, 50, 60])

# 选择元素
c = np.where(a > 3, b, a)

print(c)

输出结果为：

[ 0  1  2  3 50 60]

在上面的代码中，当 a 数组中的元素大于 3 时，选择 b 数组中的元素，否则选择 a 数组中的元素。