numpy中where函数的用法是什么

numpy中的where函数用于根据给定的条件从原数组中选择元素。它的用法如下：

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

参数：

• condition：需要满足的条件，可以是一个布尔数组或布尔表达式。
• x和y：可选参数，表示满足条件的元素替换为x中的元素，不满足条件的元素替换为y中的元素。x和y可以是一个数组或标量。

返回值：

• 返回一个新的数组，形状与原数组相同。

示例：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 3, arr, 0)
print(new_arr)
# 输出：[0 0 0 4 5]

在上面的示例中，arr > 3 是一个布尔数组，满足条件的元素为True，不满足条件的元素为False。根据条件，将满足条件的元素替换为arr中的对应元素，不满足条件的元素替换为0，得到了新的数组new_arr。