Question

numpy 两个一维数组的差集？

如a (array([1, 2, 6], dtype=int64),)
b (array([1, 2, 4, 8], dtype=int64),)
求在a中有b中没有的结果

Answer 1

#py2.7
#coding=utf-8
import numpy as np
a = np.array([1, 2, 6],dtype=np.int64)
b = np.array([1, 2, 4, 8],dtype=np.int64)
print list(set(a)-set(b))
>>> 
[6]
>>>

Answer 2

set(a)-set(b)

Answer 3

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

Answer 4

>>> import numpy as np
>>> a = np.array([1, 2, 6])
>>> b = np.array([1, 3, 6])
>>> ### 交集
>>> np.intersect1d(a,b)
array([1, 6])
>>> ### 差集
>>> np.setdiff1d(a,b)
array([2])