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

要检查NumPy数组中是否存在指定的值，可以使用np.isin()函数。该函数返回一个布尔数组，数组中的每个元素都是原数组中对应元素是否与指定值相等的结果。

下面是使用np.isin()函数的方法：

1. 导入NumPy库，创建一个NumPy数组。

import numpy as np

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

1. 使用np.isin()函数检查是否存在指定的值。

result = np.isin(arr, [3, 6])

在这个例子里，我们检查arr数组中是否存在值为3或者6的元素。执行后，会生成一个布尔数组result，其中result[i]表示arr[i]是否等于3或6。

完整代码如下：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = np.isin(arr, [3, 6])
print(result) # 输出 [False False  True False False]

1. 查看检查结果。

在上面的例子中，可以看到result的结果为[False False True False False]，这是因为arr数组中存在一个元素为3。如果我们只想知道是否存在指定的值，可以使用np.any()函数判断是否存在True值。

if np.any(result):
    print('指定的值存在于数组中')
else:
    print('指定的值不存在于数组中')

完整代码如下：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = np.isin(arr, [3, 6])
if np.any(result):
    print('指定的值存在于数组中')
else:
    print('指定的值不存在于数组中')

如果数组中存在指定的值，输出结果为指定的值存在于数组中。

示例2：

下面是一个二维数组的例子，使用np.isin()函数检查数组中是否存在指定的值：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
result = np.isin(arr, [3, 6])
print(result) # 输出 [[False False  True] [False False  True] [False False False]]

在这个例子中，np.isin()函数检查arr数组中是否存在值为3或者6的元素，生成一个与原数组大小相同的布尔数组。

我们可以使用np.any()函数对所有元素进行检查，判断指定的值是否存在于数组中。

if np.any(result):
    print('指定的值存在于数组中')
else:
    print('指定的值不存在于数组中')

完整代码如下：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
result = np.isin(arr, [3, 6])
if np.any(result):
    print('指定的值存在于数组中')
else:
    print('指定的值不存在于数组中')

如果数组中存在指定的值，输出结果为指定的值存在于数组中。