寻找两个NumPy数组之间的共同值可以使用numpy.intersect1d()函数。该函数返回两个数组之间的共同元素,这些元素已经排序并去重。
下面是一个完整的攻略:
- 导入numpy模块
在Python程序中,要使用NumPy模块的功能,首先需要导入NumPy模块。
python
import numpy as np
- 创建两个数组
在本次攻略中,我们需要创建两个数组。这可以使用numpy.array()函数来完成。在这里我们创建两个有些不同的数组。
python
a = np.array([1, 2, 3, 4, 5])
b = np.array([2, 3, 5, 7, 11])
- 调用numpy.intersect1d()函数
我们可以使用numpy.intersect1d()函数来寻找两个数组之间的共同元素。
python
c = np.intersect1d(a, b)
- 输出共同元素
为了检查这个函数返回的结果,我们可以打印出它们来。
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()函数能够在非常简洁的时间内得到我们寻找的结果。