np.where的功能

np.where()是numpy库中的一个函数，它的作用是根据指定的条件返回输入数组中满足条件的元素的索引或者元素本身。具体来说，np.where(condition, x, y)的作用是：当condition为True时，返回x中对应位置的元素；当condition为False时，返回y中对应位置的元素。

举个例子，假设有一个一维数组a，我们想要将其中大于0的元素替换成1，小于等于0的元素替换成-1，可以使用np.where()函数实现：

import numpy as np

a = np.array([1, -2, 3, 0, -4])
b = np.where(a > 0, 1, -1)
print(b)  # 输出 [-1 -1  1 -1 -1]

在上面的例子中，我们将条件设为a > 0，即大于0的元素为True，小于等于0的元素为False。当条件为True时，返回1；当条件为False时，返回-1。