Python代码列表求并集,交集,差集

目录

• 一、列表求并集

• 1. union_by

• 二、列表求交集

• 1. intersection_by

• 三、列表求差集

• 1. difference

• 2. difference_by

• 3. symmetric_difference_by

一、列表求并集

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

本篇阅读的代码片段来自于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时，函数转化为直接求取两个列表的对称差。

来源：https://segmentfault.com/a/1190000040807166