Python中如何用一行代码实现两个列表的去重交集计算?
一行代码实现方案
你可以直接在现有代码的基础上,把过滤后的结果先转成集合去重,再转回列表即可,修改后的写法如下:
def intersection(list1, list2): return list(set(filter(lambda x: x in list2, list1)))
也可以用更简洁的集合交集写法,运行效率更高:
def intersection(list1, list2): return list(set(list1) & set(list2))
如果需要保留元素在第一个列表中首次出现的顺序(Python 3.7及以上版本原生支持),可以用如下保序写法:
def intersection(list1, list2): return list(dict.fromkeys(x for x in list1 if x in list2))
效果验证
调用intersection([1,2,3,3,4], [8,3,9])时,以上三种写法都会返回正确结果[3],不存在重复元素。
内容的提问来源于stack exchange,提问作者natitati
相关产品推荐
相关产品推荐

