np.where怎么用

np.where()函数是NumPy中的一个函数，用于在满足特定条件的情况下返回相应的值。np.where()的语法如下：

np.where(condition, x, y)

其中，condition是一个布尔型的NumPy数组或逻辑表达式，x和y则是两个数组，它们的形状必须相同。当condition中的元素为True时，返回x中对应位置的元素，否则返回y中对应位置的元素。

下面是一个简单的例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([10, 20, 30, 40, 50])

c = np.where(a > 3, b, a)

print(c)

输出：

[ 1  2  3 40 50]

在上面的例子中，np.where()函数的条件是a > 3，因此在a中大于3的位置，返回b中对应位置的元素，否则返回a中对应位置的元素。因此，c中的第4和第5个元素是40和50，即b中大于3的位置的元素，而其他元素则是a中对应位置的元素。