python中np.where函数

np.where函数是numpy库中的一个函数，它的作用是根据指定的条件，返回符合条件的元素的索引或值。np.where函数的语法如下：

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

其中，condition是指定的条件，x和y是可选的参数，用于指定符合条件和不符合条件的返回值。

当只有condition参数时，np.where函数返回符合条件的元素的索引。

当有x和y参数时，np.where函数会根据condition的值，返回符合条件的元素的值或不符合条件的元素的值。

例如，以下代码会将数组a中小于0的元素替换为0，其他元素不变：

import numpy as np

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

b = np.where(a < 0, 0, a)

print(b)

输出结果为：

[0 2 0 4 5]

其中，np.where(a < 0, 0, a)的作用是将数组a中小于0的元素替换为0，其他元素不变。