如何检查NumPy数组中是否存在指定的值

  • Post category:Python

检查NumPy数组中是否存在指定的值可以通过numpy.isin()函数来实现。该函数返回一个布尔值数组,数组中的每个元素都代表原始数组是否包含指定值。

使用numpy.isin()函数的语法如下:

numpy.isin(element, test_elements, assume_unique=False, invert=False)

其中,element为要检查的值;test_elements为要被检查的数组;assume_unique参数表示被检查的数组是否唯一,若为True则执行更快;invert参数表示是否对结果进行反转(True则返回原数组不包含元素的位置的值)。

以下是两个具体的示例说明:

示例1:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
value = 3
result = np.isin(value, arr)

print(result)

运行结果如下:

True

示例2:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
values = [3, 6]
result = np.isin(values, arr)

print(result)

运行结果如下:

[ True False]

以上就是检查NumPy数组中是否存在指定值的完整攻略。通过numpy.isin()函数可以方便地检查指定的数值是否在数组中,也可以检查一些复杂的条件。