Python中的np.setdiff1d()函数

Python中的np.setdiff1d()函数可用于找出两个序列集合中元素的差异

API及参数如下：

np.setdiff1d(ar1,ar2,assume_unique)：找出序列1在序列2中的差异，并返回序列1中不在序列2中的元素

• ar1：输入数组
• ar2：输入比较数组
• assume_unique：是否假设唯一，默认为False，输出去重且元素从小到大排序，如果为True，则不会去重，输出顺序保持原来的顺序

以下是一个使用示例：

import numpy as np

a = np.array([3, 2, 2, 1])
b = np.array([5, 4, 3, 3])
# a = [3, 2, 2, 1]
# b = [5, 4, 3, 3]

c = np.setdiff1d(a, b)
print(c)    # [1 2]

c = np.setdiff1d(a, b, assume_unique=True)
print(c)    # [2 2 1]