可以使用 np.intersect1d()
函数
来找到两个数组中相同的元素。示例如下:
import numpy as np
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([2, 4, 6, 8, 10])
same_elements = np.intersect1d(arr1, arr2)
print('The same elements are:', same_elements)
The same elements are: [2 4]
在上面的代码中,我们首先导入了 NumPy 模块。然后,我们定义了两个一维数组 arr1 和 arr2,并将它们赋给相应的变量。
接下来,我们使用 np.intersect1d() 函数来查找这两个数组中相同的元素,并将其赋给变量 same_elements。最后,我们打印出了相同的元素。
需要注意的是,np.intersect1d() 函数返回的是一个排序后的数组,其中只包含唯一值。如果两个输入数组包含重复值,则这些值仅会出现在输出数组中的一个位置。
如果要找到两个数组中所有重复的元素,可以使用以下代码:
import numpy as np
arr1 = np.array([1, 2, 3, 2, 4, 5])
arr2 = np.array([2, 4, 6, 8, 10, 2])
same_elements = np.intersect1d(arr1, arr2, assume_unique=False)
print('The same elements are:', same_elements)
The same elements are: [2 4]
在这个例子中,我们在 arr1 和 arr2 中加入了重复的元素,然后使用 assume_unique 参数将其设置为 False,以确保输出数组包含重复的元素。