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

  • Post category:Python

寻找两个NumPy数组之间的共同值,我们可以借助Python中的集合操作完成。下面是完整攻略:

  1. 将两个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)
  1. 使用集合操作寻找共同的值

在Python中,可以使用集合操作符来完成对集合的交集、并集、差集等操作。在这里,我们只需要使用交集操作符&来计算两个集合的共同部分即可。代码如下:

intersection = set_a & set_b
  1. 将交集转换为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数组之间的共同值的完整攻略。