Python列表求并集,交集,差集 Python代码列表求并集,交集,差集
Felix 人气:0一、列表求并集
实现了两个列表求并集的功能。同时支持使用一个过滤条件函数,列表中所有元素根据该条件求取并集,并集中是两个列表的原始元素。
本篇阅读的代码片段来自于30-seconds-of-python。
1. 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应用于所有的列表元素上。
set
是Python
的一个特殊的数据类型,是由不重复元素组成的无序的集。本函数直接使用set
类型消除了列表中的重复元素。
特别的:当过滤条件函数是lamda x:x时,函数转化为直接求取两个列表的并集。
二、列表求交集
实现了两个列表求交集的功能。同时支持使用一个过滤条件函数,列表中所有元素根据该条件求取交集,交集中是两个列表的原始元素。
1. intersection_by
def intersection_by(a, b, fn): _b = set(map(fn, b)) return [item for item in a if fn(item) in _b] # EXAMPLES from math import floor intersection_by([2.1, 1.2], [2.3, 3.4],floor) # [2.1]
intersection_by
函数接收两个列表和一个过滤条件函数。将提供的函数应用于两个列表中的每个元素后,返回一个存在于两个列表中的元素的列表。函数通过对b中的每个元素应用fn来创建一个集合,然后在a上结合使用列表推导式和fn
来只保留同时存在于两个列表中的值。
map
函数之前已经讲解过,它会返回一个迭代器,该迭代器会将变换函数fn应用于所有的列表元素上。
set是Python
的一个特殊的数据类型,是由不重复元素组成的无序的集。本函数直接使用set
类型消除了列表中的重复元素。
特别的:当过滤条件函数是lamda x:x时,函数转化为直接求取两个列表的交集。
三、列表求差集
实现三种不同的列表差集的实现方式。分别是列表直接求差difference
,列表中所有元素根据条件函数求差difference_by
,列表中所有元素根据条件函数求对称差symmetric_difference_by
。
1. difference
def difference(a, b): _b = set(b) return [item for item in a if item not in _b] # EXAMPLES difference([1, 2, 3], [1, 2, 4]) # [3]
difference
函数返回两个可迭代对象之间的差。该函数从b中创建一个集合_b
,然后在a上使用列表推导式,只保留_b中不包含的值。该函数中a和b是有顺序关系的,a-b是从a中删除b中包含的数据。
set
是Python
的一个特殊的数据类型,是由不重复元素组成的无序的集。本函数直接使用set类型消除了列表中的重复元素。
2. difference_by
def difference_by(a, b, fn): _b = set(map(fn, b)) return [item for item in a if fn(item) not in _b] # EXAMPLES from math import floor difference_by([2.1, 1.2], [2.3, 3.4], floor) # [1.2] difference_by([{ 'x': 2 }, { 'x': 1 }], [{ 'x': 1 }], lambda v : v['x']) # [ { x: 2 } ]
difference_by
函数接收两个列表和一个过滤条件函数。将提供的函数应用于两个列表中的每个元素后,返回两个原始列表的差。函数通过对b中的每个元素应用fn来创建一个集合,然后在a上使用列表推导式与fn相结合,只保留之前创建的集合_b中没有包含的值。
特别的:当过滤条件函数是lamda x:x时,函数转化为直接求取两个列表的差。
3. symmetric_difference_by
def symmetric_difference_by(a, b, fn): _a, _b = set(map(fn, a)), set(map(fn, b)) return [item for item in a if fn(item) not in _b] + [item for item in b if fn(item) not in _a] # EXAMPLES from math import floor symmetric_difference_by([2.1, 1.2], [2.3, 3.4],floor) # [1.2, 3.4]
symmetric_difference_by
函数对两个列表中的每个列表元素应用所提供的函数后,返回两个原始列表之间的对称差。函数通过分别对每个列表中的每个元素应用fn来创建两个集合_a和_b,然后在每个元素上使用列表理解与fn相结合,只保留不包含在之前创建的其他集合中的值(在a 中,不在_b中;在b中,不在_a中。)。
特别的:当过滤条件函数是lamda x:x时,函数转化为直接求取两个列表的对称差。
加载全部内容