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

时间:2022-09-01 15:05:27

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

一、列表求并集

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

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

1. union_by

?
1
2
3
4
5
6
7
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时,函数转化为直接求取两个列表的并集。

二、列表求交集

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

1. intersection_by

?
1
2
3
4
5
6
7
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

?
1
2
3
4
5
6
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中包含的数据。

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

2. difference_by

 

?
1
2
3
4
5
6
7
8
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

?
1
2
3
4
5
6
7
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时,函数转化为直接求取两个列表的对称差。

到此这篇关于Python代码列表求并集,交集,差集的文章就介绍到这了,更多相关Python列表求并集,交集,差集内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://segmentfault.com/a/1190000040807166