博客

寻找两个NumPy数组之间的共同值的方法有多种，以下是其中两种方法的示例说明：

方法一：使用intersect1d函数

intersect1d()函数可以用来找到两个NumPy数组之间共同的元素。这个函数输入两个数组作为参数，然后返回一个新的排序数组，其中包含两个输入数组中共有的元素。代码示例如下：

import numpy as np

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

common_arr = np.intersect1d(arr1, arr2)

print("Common elements between arr1 and arr2")
print(common_arr)

输出结果如下：

Common elements between arr1 and arr2
[3 4 5]

方法二：使用in1d函数

in1d()函数可以用来找到一个NumPy数组中的元素是否包含在另一个数组中。代码示例如下：

import numpy as np

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

common_arr = arr1[np.in1d(arr1, arr2)]

print("Common elements between arr1 and arr2")
print(common_arr)

输出结果如下：

Common elements between arr1 and arr2
[3 4 5]

以上两种方法都可以用来寻找两个NumPy数组之间共同的元素，具体选择哪种方法取决于你的具体需求。