Python numpy.where()

numpy.where() 方法是 Python 中的一个非常有用的函数，它允许我们根据特定的条件在数组中进行元素级别的选择操作。

用法

numpy.where(condition, x, y) 的参数说明如下：

• condition：一个返回布尔值的数组或标量。具有 True 值的元素将被选中。
• x：一个与 condition 同样大小的数组或标量，其中对应于 True 值的元素将被选择。
• y：一个与 condition 同样大小的数组或标量，其中对应于 False 值的元素将被选择。

numpy.where() 返回一个新的数组，该数组包含根据条件选择的元素。

示例

让我们通过一个示例来看看 numpy.where() 的用法：

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

condition = arr > 5
result = np.where(condition, arr, 0)

print(result)

输出结果如下所示：

[[0 0 0]
 [0 0 6]
 [7 8 9]]

在上面的示例中，我们创建了一个二维数组 arr。然后，我们定义了一个条件 condition，该条件指定了要选择哪些元素。我们使用 arr > 5 来创建一个与 arr 相同大小的布尔数组，其中 True 表示 arr 的元素大于 5，False 表示 arr 的元素小于等于 5。

然后，我们使用 np.where(condition, arr, 0) 来选择符合条件的元素。如果条件为 True，则选择 arr 对应位置的元素；如果条件为 False，则选择 0。

最后，我们打印出结果数组 result，它包含了根据条件选择的元素。

总结

numpy.where() 是一个非常有用的函数，它允许我们根据条件选择数组中的元素。它的灵活性使得我们能够根据我们的需求对数组进行选择和操作，从而简化了我们的编程工作。