numpy.where

`numpy.where` 是一个用于在数组中根据给定条件返回元素的索引的函数。

函数语法如下：

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

参数说明：

- `condition`：一个布尔型或非零的数组，用于指定筛选条件。
- `x`：可选参数，当 `condition` 中的元素为 `True` 时，返回的数组中对应位置的元素值，缺省值为 `None`。
- `y`：可选参数，当 `condition` 中的元素为 `False` 时，返回的数组中对应位置的元素值，缺省值为 `None`。

函数返回值：

`numpy.where` 函数返回一个元组，其中包含满足条件的元素的索引。如果 `x` 和 `y` 都没有给定，则返回一个包含索引的元组；如果 `x` 和 `y` 都给定，则返回一个根据 `condition` 条件计算的新数组。

以下是一个示例代码：

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])

c = np.where(condition, a, b)
print(c)

输出结果为：

[ 1 20  3 40  5]

在上面的例子中，我们创建了两个数组 `a` 和 `b`，以及一个布尔型数组 `condition`。然后使用 `numpy.where` 函数根据 `condition` 数组中的值选择 `a` 或 `b` 中的对应值，最终生成一个新的数组 `c`。