Python中的 numpy.where()

在NumPy中，numpy.where()函数是用于返回满足给定条件的元素的索引。它的一般语法如下：

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

其中：

• condition：表示需要满足的条件。
• x：表示满足条件的索引的值。
• y：表示未满足条件的索引的值。

示例

下面我们来介绍一些具体的例子。

示例 1

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
arr2 = np.array([10, 20, 30, 40, 50, 60, 70, 80, 90, 100])

result = np.where(arr1 > 5, arr1, arr2)

print(result)

输出：

[10 20 30 40 50  6  7  8  9 10]

示例 2

import numpy as np

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

result = np.where(arr == 0)

print(result)

输出：

(array([3, 5, 9], dtype=int64),)

示例 3

import numpy as np

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

result = np.where(arr > 5, 'yes', 'no')

print(result)

输出：

['no' 'no' 'no' 'no' 'no' 'no' 'yes' 'yes' 'yes' 'no']

总结

numpy.where()函数是NumPy中用于返回满足给定条件的元素的索引的函数。它可以通过给定condition参数来指定需要满足的条件，并返回满足条件的值或索引。我们可以通过这个函数来对数组进行操作。