Numpy中单个元素是否包含在Numpy数组中的检查方法

在本文中，我们将介绍如何通过使用Numpy库中的函数来检查Numpy数组中是否包含特定元素。

阅读更多：Numpy 教程

什么是Numpy？

Numpy是在Python中进行科学计算时使用的常用库。它提供各种功能，包括矩阵和数组的操作，以及用于数学、科学和工程应用的函数。

Numpy数组

Numpy数组是一个由同类型数据组成的集合。它们可以是一维数组或多维数组。以下是一个由Python列表转换成的一维Numpy数组示例：

import numpy as np

mylist = [1, 2, 3, 4, 5]
myarray = np.array(mylist)
print(myarray)

输出结果：

[1 2 3 4 5]

以下是一个由Python列表转换成的二维Numpy数组示例：

import numpy as np

mylist = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
myarray = np.array(mylist)
print(myarray)

输出结果：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Numpy中的in1d函数

Numpy中的in1d函数可以用于检查一个数组中的元素是否包含在另一个数组中。如果元素包含在数组中，函数将返回一个True值，否则返回False值。

以下是一个将列表转换成Numpy数组并使用in1d函数检查元素是否包含在数组中的示例：

import numpy as np

mylist = [1, 2, 3, 4, 5]
myarray = np.array(mylist)
print(np.in1d(3, myarray))
print(np.in1d(6, myarray))

输出结果：

True
False

指定位置的检查

当我们想要检查一个元素是否包含在Numpy数组的特定位置时，我们可以使用in1d函数的另一个变体：
“`np.where“`。

np.where函数将返回指定元素的位置。

以下是一个使用np.where函数来检查包含在特定位置的元素：

import numpy as np

mylist = [1, 2, 3, 4, 5]
myarray = np.array(mylist)
print(np.where(myarray == 3))
print(np.where(myarray == 6))

输出结果：

(array([2]),)
(array([], dtype=int64),)

这些输出结果可以解读为：

• 当Numpy数组中包含元素3时，它的位置是2。
• 当Numpy数组中不包含元素6时，该位置为空。

总结

Numpy是一个强大的工具，可以用于科学、工程和数学运算。在本文中，我们了解了如何使用Numpy库中的函数来检查单个元素是否包含在一个Numpy数组中，以及如何指定元素的位置。希望这篇文章能够帮助你更好地理解Numpy的功能。