python中np.where()

np.where()函数是NumPy库中的一个函数，用于根据一定条件返回一个数组中符合条件的元素的索引或元素本身。

语法：

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

参数：

condition：表示条件，可以是一个布尔数组、一个条件表达式或一个函数，如果为True则返回x数组中对应的元素，否则返回y数组中对应的元素。

x：表示符合条件的元素将从x数组中返回，如果x数组没有被给定，则返回满足条件的元素的索引。

y：表示不符合条件的元素将从y数组中返回，如果y数组没有被给定，则返回满足条件的元素的索引。

返回值：

返回一个元组，其中包含满足条件的元素的索引或元素本身，具体返回值取决于参数x和y是否被给定。

示例：

import numpy as np

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

print(np.where(condition, a, b))
# 输出：array([ 1, 20,  3, 40,  5])

在上面的示例中，我们定义了两个数组a和b，以及一个条件数组condition。np.where()函数根据条件数组中的True和False，对a和b数组中的元素进行选择，返回一个新的数组。具体来说，当condition数组中对应位置为True时，选择a数组中的元素，否则选择b数组中的元素。最终得到的数组为[1, 20, 3, 40, 5]。