寻找两个NumPy数组之间的共同值,我们可以借助Python中的集合操作完成。下面是完整攻略:
- 将两个NumPy数组转换为Python中的集合
在Python中,使用set()函数可以将一个列表或数组转换为一个集合对象。因此,我们可以分别将两个NumPy数组转换为Python集合对象,代码如下:
import numpy as np
a = np.array([1, 2, 3, 4])
b = np.array([3, 4, 5, 6])
set_a = set(a)
set_b = set(b)
- 使用集合操作寻找共同的值
在Python中,可以使用集合操作符来完成对集合的交集、并集、差集等操作。在这里,我们只需要使用交集操作符&
来计算两个集合的共同部分即可。代码如下:
intersection = set_a & set_b
- 将交集转换为NumPy数组
将交集转换为NumPy数组,可以使用numpy.array()
函数。代码如下:
common = np.array(list(intersection))
# list(intersection)将交集转换为列表
下面是两条示例说明:
示例一:
a = np.array([1, 2, 3, 4])
b = np.array([3, 4, 5, 6])
set_a = set(a)
set_b = set(b)
intersection = set_a & set_b
common = np.array(list(intersection))
print(common)
执行以上代码,输出结果为:
[3 4]
示例二:
a = np.array([1, 3, 3, 5, 7])
b = np.array([2, 3, 4, 6, 7])
set_a = set(a)
set_b = set(b)
intersection = set_a & set_b
common = np.array(list(intersection))
print(common)
执行以上代码,输出结果为:
[3 7]
以上就是寻找两个NumPy数组之间的共同值的完整攻略。