Python中的np.setdiff1d()
函数可用于找出两个序列集合中元素的差异
API及参数说明如下:
np.setdiff1d(ar1,ar2,assume_unique):找出序列1在序列2中的差异,并返回序列1中不在序列2中的元素
- ar1:输入数组
- ar2:输入比较数组
- assume_unique:是否假设唯一,默认为False,输出去重且元素从小到大排序,如果为True,则不会去重,输出顺序保持原来的顺序
以下是一个示例:
import numpy as npa = 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]