python求交集、并集、差集

1. 获取两个list 的交集元素:

a = [1, 2, 3, 4, 1 ,2 ,3 ,4]
b = [3, 4, 5, 6, 5, 6]

#方法一:

print([i for i in set(a) if i in set(b)])

结果: [3, 4]
#方法二,比方法一快很多!

print(set(a).intersection(b))

结果: {3, 4}

2. 获取两个list 的并集元素:

print(set(a).union(b))

结果:{1, 2, 3, 4, 5, 6}

3. 获取两个 list 的差集元素:

print(set(a).difference(b))  # a中有而b中没有的非常高效!

结果:{1, 2}

print(set(b).difference(a))  # b中有而a中没有

结果:{5, 6}

切记使用set返回值为集合类型,不考虑重复元素,目的为取交集、并集、差集的元素

你可能感兴趣的:(python,python,开发语言,后端)