当我们需要寻找两个NumPy数组之间的共同值时,可以使用NumPy提供的intersect1d函数来实现。
intersect1d函数可以计算两个数组的交集,返回一个由交集元素组成的新数组。
下面是详细的步骤:
1.导入NumPy库
import numpy as np
2.创建两个NumPy数组,假设它们为arr1和arr2
arr1 = np.array([2, 3, 4, 5, 6, 7])
arr2 = np.array([5, 6, 7, 8, 9])
3.使用intersect1d函数计算两个数组的交集
common_elements = np.intersect1d(arr1, arr2)
4.打印结果
print(common_elements)
上述代码将输出共同值为5、6、7的新数组。
下面是另一个示例:
arr1 = np.array(['apple', 'banana', 'cherry'])
arr2 = np.array(['banana', 'cherry', 'durian'])
common_fruits = np.intersect1d(arr1, arr2)
print(common_fruits)
上述代码将输出共同水果为banana和cherry的新数组。
注意:在使用intersect1d函数时,两个数组可以是任何形状和尺寸。仅仅只需要把它们作为参数传递即可。