检查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()
函数可以方便地检查指定的数值是否在数组中,也可以检查一些复杂的条件。