Python代碼列表求並集,交集,差集

一、列表求並集

實現瞭兩個列表求並集的功能。同時支持使用一個過濾條件函數,列表中所有元素根據該條件求取並集,並集中是兩個列表的原始元素。

本篇閱讀的代碼片段來自於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應用於所有的列表元素上。

setPython的一個特殊的數據類型,是由不重復元素組成的無序的集。本函數直接使用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中包含的數據。

setPython的一個特殊的數據類型,是由不重復元素組成的無序的集。本函數直接使用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時,函數轉化為直接求取兩個列表的對稱差。

到此這篇關於Python代碼列表求並集,交集,差集的文章就介紹到這瞭,更多相關Python列表求並集,交集,差集內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: