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

  • Post category:Python

当我们需要寻找两个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函数时,两个数组可以是任何形状和尺寸。仅仅只需要把它们作为参数传递即可。