Python中的 numpy.intersect1d()函数

在 Python 的 numpy 模块中，有一个很有用的函数叫做 intersect1d()。该函数用于对两个数组或者序列进行比较，并返回它们的交集，即同时出现在两个数组或者序列中的元素。在数据分析和处理中，该函数可以用于查找两个数据集中的共同元素。

函数语法

numpy.intersect1d(ar1, ar2, assume_unique=False, return_indices=False)

参数说明

• ar1, ar2：需要比较的两个数组或序列
• assume_unique：可选参数，如果设置成 True，则假设 ar1 和 ar2 中的元素都是唯一的。默认值为 False，即无任何假设。
• return_indices：可选参数，如果设置成 True，则返回 ar1 和 ar2 中元素的下标。默认值为 False。

返回值

该函数返回两个数组的交集，即同时出现在 ar1 和 ar2 中的元素。如果设置参数 return_indices 为 True，则还会返回 ar1 和 ar2 中元素的下标。

例子

import numpy as np

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

# 查找两个数组的交集
c = np.intersect1d(a, b)

print(c)  # 输出为 [3]

在上面的例子中，我们定义了两个数组 a 和 b。然后，我们使用 intersect1d() 函数查找两个数组的交集，并将结果赋值给变量 c。最后，我们打印出变量 c 的值，即 [3]。这是因为 3 同时出现在 a 和 b 两个数组中。

小结

在本篇文章中，我们介绍了 numpy 模块中的 intersect1d() 函数，其用途是查找两个数组或者序列中的交集。通过本文的例子，您可以学会如何使用该函数，并将其应用到实际数据处理中。