博客

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

  • Post category:Python

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

np.isin()函数的语法如下：

np.isin(element_to_find, array, assume_unique=False, invert=False)
  • element_to_find：要查找的元素
  • array：待检查的数组
  • assume_unique：如果为True，则假定输入数组已经是唯一的（无重复值），来提高性能。
  • invert：如果为True，则返回逐元素逆的布尔值。

该函数返回一个与输入数组大小相同的布尔数组，在该数组中，对应元素的值为True表示对应位置上的元素等于要查找的值，否则为False。

下面是两个示例说明。

示例1：检查一维数组中是否存在指定的值

import numpy as np

# 创建一维数组
arr = np.array([1, 3, 5, 7, 9])

# 检查数组中是否存在3
if np.isin(3, arr):
    print("数组中存在3。")
else:
    print("数组中不存在3。")

# 检查数组中是否存在4
if np.isin(4, arr):
    print("数组中存在4。")
else:
    print("数组中不存在4。")

输出结果：

数组中存在3。
数组中不存在4。

示例2：检查二维数组中是否存在指定的值

import numpy as np

# 创建二维数组
arr = np.array([[1, 2], [3, 4], [5, 6]])

# 检查数组中是否存在2
if np.isin(2, arr):
    print("数组中存在2。")
else:
    print("数组中不存在2。")

# 检查数组中是否存在7
if np.isin(7, arr):
    print("数组中存在7。")
else:
    print("数组中不存在7。")

输出结果：

数组中存在2。
数组中不存在7。

以上就是检查NumPy数组中是否存在指定的值的详细攻略，希望对你有所帮助。

Tags: