List取交集的多个方面详解

一、基础使用

1、用Python自带的set函数实现列表取交集

a = [1, 2, 3, 4, 5]
b = [3, 4, 5, 6, 7]
c = set(a) & set(b)
print(list(c))
# 返回结果为 [3, 4, 5]

2、用Python自带的intersection函数实现列表取交集

a = [1, 2, 3, 4, 5]
b = [3, 4, 5, 6, 7]
c = list(set(a).intersection(set(b)))
print(c)
# 返回结果为 [3, 4, 5]

3、用Python自带的filter函数实现列表取交集

a = [1, 2, 3, 4, 5]
b = [3, 4, 5, 6, 7]
c = list(filter(lambda x: x in a, b))
print(c)
# 返回结果为 [3, 4, 5]

二、列表里嵌套列表

1、普通的嵌套列表

a = [[1, 2], [2, 3], [3, 4]]
b = [[2, 3], [3, 4], [4, 5]]
c = list(filter(lambda x: x in a, b))
print(c)
# 返回结果为 [[2, 3], [3, 4]]

2、列表里元素也是列表

a = [[[1, 2], [2, 3]], [[3, 4], [4, 5]], [[5, 6], [6, 7]]]
b = [[[2, 3], [3, 4]], [[4, 5], [5, 6]], [[6, 7], [7, 8]]]
c = []
for i in a:
    for j in b:
        if i == j:
            c.append(i)
print(c)
# 返回结果为 [[[2, 3], [3, 4]], [[6, 7], [7, 8]]]

三、字典类型中取交集

1、字典类型的键值相交

a = {'a': 1, 'b': 2, 'c': 3}
b = {'b': 4, 'c': 3, 'd': 5}
c = list(a.keys() & b.keys())
print(c)
# 返回结果为 ['c', 'b']

2、字典类型的值相交

a = {'a': 1, 'b': 2, 'c': 3}
b = {'d': 5, 'e': 3, 'f': 2}
c = list(set(a.values()).intersection(set(b.values())))
print(c)
# 返回结果为 [2, 3]

四、保留重复元素

1、用Counter函数统计交集元素出现的次数

from collections import Counter

a = [1, 2, 2, 3, 4]
b = [2, 2, 3, 3, 5]
c = list((Counter(a) & Counter(b)).elements())
print(c)
# 返回结果为 [2, 2, 3]

2、用set集合替换Counter函数，实现更高效的操作

a = [1, 2, 2, 3, 4]
b = [2, 2, 3, 3, 5]
c = list((set(a) & set(b)))
print(c)
# 返回结果为 [2, 3]

五、不同数据类型之间取交集

1、列表与集合取交集

a = [1, 2, 3, 4, 5]
b = {3, 4, 5, 6, 7}
c = list(set(a).intersection(b))
print(c)
# 返回结果为 [3, 4, 5]

2、字符串与列表取交集

a = 'abcdefghijklmnopqrstuvwxyz'
b = ['a', 'e', 'i', 'o', 'u']
c = list(set(a).intersection(b))
print(c)
# 返回结果为 ['a', 'e', 'i', 'o', 'u']