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

  • Post category:Python

寻找两个NumPy数组之间的共同值,可以使用np.intersect1d()函数来实现。该函数的作用是获取两个一维数组中的公共元素值,返回的是一个新的一维数组。

以下是具体步骤:

1.导入NumPy库:

import numpy as np

2.准备两个NumPy数组:

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

3.使用np.intersect1d()函数计算共同值:

common = np.intersect1d(a, b)

4.打印共同值:

print(common)

输出结果如下:

[3 4 5]

代码示例:

import numpy as np

# 准备两个NumPy数组
a = np.array([1, 2, 3, 4, 5])
b = np.array([3, 4, 5, 6, 7])

# 计算共同值
common = np.intersect1d(a, b)

# 打印结果
print(common)

输出结果:

[3 4 5]

另外一个示例:

import numpy as np

# 准备两个NumPy数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[3, 4], [5, 6]])

# 计算共同值
common = np.intersect1d(a, b)

# 打印结果
print(common)

输出结果:

[3 4]

以上就是使用np.intersect1d()函数寻找两个NumPy数组之间的共同值的完整攻略。