一、列表求并集
實現了兩個列表求并集的功能。同時支持使用一個過濾條件函數,列表中所有元素根據該條件求取并集,并集中是兩個列表的原始元素。
本篇閱讀的代碼片段來自于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應用于所有的列表元素上。
set
是Python
的一個特殊的數據類型,是由不重復元素組成的無序的集。本函數直接使用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中包含的數據。
set
是Python
的一個特殊的數據類型,是由不重復元素組成的無序的集。本函數直接使用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