博客

检查NumPy数组中是否存在指定的值，可以使用np.isin()方法。

np.isin(x, vals, assume_unique=False, invert=False)

x：待检查的数组

vals：要检查的值或者列表

assume_unique：是否已经去重。如果设为True，函数运行速度会更快。

invert：是否取反。如果设为True，则返回一个与输入数组形状相同的布尔值数组，并且数组中所有给定的值都设置为False。

下面给出两个示例说明：

1.检查数组中是否存在单个值

import numpy as np

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

print(np.isin(arr, 3))

输出：

[False False True False False]

2.检查数组中是否存在多个值

import numpy as np

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

print(np.isin(arr, [3, 4, 6]))

输出：

[False False  True  True False]

以上两个示例展示了如何使用np.isin()方法来检查NumPy数组中是否存在指定的值。需要注意的是，np.isin()返回的是一个布尔类型的数组，数组中值为True表示对应的元素在指定的值中存在，否则表示不存在。