Python代码阅读(第14篇):列表求并集

2021年10月15日 6点热度 0条评论 来源: FelixZ

Python 代码阅读合集介绍:为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现了实现了两个列表求并集的功能。同时支持使用一个过滤条件函数,列表中所有元素根据该条件求取并集,并集中是两个列表的原始元素。

本篇阅读的代码片段来自于30-seconds-of-python

union_by

def union_by(a, b, fn):
_a = set(map(fn, a))
return list(set(a + [item for item in b if fn(item) not in _a]))

# EXAMPLES
from math import floor
union_by([2.1], [1.2, 2.3], floor) # [2.1, 1.2]

union_by函数接收两个列表和一个过滤条件函数。将提供的函数应用于两个列表中的每个元素后,返回一个新的列表,包含所有存在于两个列表的不重复的元素。

map函数之前已经讲解过,它会返回一个迭代器,该迭代器会将变换函数fn应用于所有的列表元素上。

setPython的一个特殊的数据类型,是由不重复元素组成的无序的集。本函数直接使用set类型消除了列表中的重复元素。

特别的,当过滤条件函数是lamda x:x时,函数转化为直接求取两个列表的并集。

    原文作者:FelixZ
    原文地址: https://www.cnblogs.com/felixz/p/15402934.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系管理员进行删除。