Python中的 numpy.setdiff1d()函数介绍

简介

numpy.setdiff1d() 函数用于找到两个数组中不同的元素。它返回一个新数组，该数组包含在第一个数组中但不在第二个数组中出现的元素。该函数还可以指定是否要返回唯一元素。

语法

numpy.setdiff1d(ar1, ar2, assume_unique=False)

参数

• ar1：第一个输入数组。
• ar2：第二个输入数组。
• assume_unique：可选项。如果设置为 True，则输入数组被视为已排序，并且仅包含唯一元素。

返回值

该函数返回输入数组的差异，类型为 ndarray。

示例

import numpy as np

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

print(result)

输出：

[1 2 3]

在上面的示例中，我们有两个数组 arr1 和 arr2。通过使用 numpy.setdiff1d() 函数，我们可以找到数组 arr1 中不在 arr2 中的元素。结果将被存储在 result 数组中，并输出到控制台上。

小结

在此文章中，我们介绍了 numpy.setdiff1d() 函数，并提供了使用该函数的示例。 numpy 库是处理数学运算和数据科学应用的最佳选择之一，该函数是其中一个非常有用的工具。