寻找两个NumPy数组之间的共同值

  • Post category:Python

寻找两个NumPy数组之间的共同值可以使用numpy.intersect1d()函数。该函数返回两个数组之间的共同元素,这些元素已经排序并去重。

下面是一个完整的攻略:

  1. 导入numpy模块

在Python程序中,要使用NumPy模块的功能,首先需要导入NumPy模块。

python
import numpy as np

  1. 创建两个数组

在本次攻略中,我们需要创建两个数组。这可以使用numpy.array()函数来完成。在这里我们创建两个有些不同的数组。

python
a = np.array([1, 2, 3, 4, 5])
b = np.array([2, 3, 5, 7, 11])

  1. 调用numpy.intersect1d()函数

我们可以使用numpy.intersect1d()函数来寻找两个数组之间的共同元素。

python
c = np.intersect1d(a, b)

  1. 输出共同元素

为了检查这个函数返回的结果,我们可以打印出它们来。

python
print(c)

下面是两个数组之间共同元素的一个示例:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([2, 3, 5, 7, 11])

c = np.intersect1d(a, b)

print(c)

其中,输出结果就是两个数组中共同的元素:

[2 3 5]

下面是第二个示例,其数组更大:

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
b = np.array([5, 6, 7, 8, 9, 10, 11, 12, 13, 14])

c = np.intersect1d(a, b)

print(c)

输出结果也就是两个数组中共同的元素:

[ 5  6  7  8  9 10]

以上就是寻找两个NumPy数组之间的共同值的攻略,使用numpy.intersect1d()函数能够在非常简洁的时间内得到我们寻找的结果。