numpy的np.where

numpy的np.where函数用于根据指定条件返回输入数组中元素的索引或值。

语法：numpy.where(condition, [x, y])

参数：
- condition：必需，表示条件，可以是一个布尔数组或可以转换为布尔数组的对象。
- x：可选，表示符合条件的元素要被替换成的值或数组。如果未指定，则返回符合条件的元素的索引。
- y：可选，表示不符合条件的元素要被替换成的值或数组。

返回值：
- 如果只有condition参数，则返回元素为True的索引。满足条件(condition)，输出x，不满足输出y。

例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([True, False, True, False, False])

# 返回条件为True的元素的索引
indices = np.where(b)
print(indices)  # 输出 (array([0, 2]),)

# 返回符合条件的元素
values = np.where(b, a, -1)
print(values)  # 输出 [1 -1 3 -1 -1]