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

更新時間：2021年11月11日 16:32:39 作者：Felix

這篇文章主要介紹了Python代碼列表求并集,交集,差集，下面文章講詳細(xì)的介紹如何利用python代碼實(shí)現(xiàn)并集,交集,差集的相關(guān)資料展開內(nèi)容,需要的朋友可以參考一下

一、列表求并集

實(shí)現(xiàn)了兩個列表求并集的功能。同時支持使用一個過濾條件函數(shù)，列表中所有元素根據(jù)該條件求取并集，并集中是兩個列表的原始元素。

本篇閱讀的代碼片段來自于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函數(shù)接收兩個列表和一個過濾條件函數(shù)。將提供的函數(shù)應(yīng)用于兩個列表中的每個元素后，返回一個新的列表，包含所有存在于兩個列表的不重復(fù)的元素。

map函數(shù)之前已經(jīng)講解過，它會返回一個迭代器，該迭代器會將變換函數(shù)fn應(yīng)用于所有的列表元素上。

set是Python的一個特殊的數(shù)據(jù)類型，是由不重復(fù)元素組成的無序的集。本函數(shù)直接使用set類型消除了列表中的重復(fù)元素。

特別的：當(dāng)過濾條件函數(shù)是lamda x:x時，函數(shù)轉(zhuǎn)化為直接求取兩個列表的并集。

二、列表求交集

實(shí)現(xiàn)了兩個列表求交集的功能。同時支持使用一個過濾條件函數(shù)，列表中所有元素根據(jù)該條件求取交集，交集中是兩個列表的原始元素。

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函數(shù)接收兩個列表和一個過濾條件函數(shù)。將提供的函數(shù)應(yīng)用于兩個列表中的每個元素后，返回一個存在于兩個列表中的元素的列表。函數(shù)通過對b中的每個元素應(yīng)用fn來創(chuàng)建一個集合，然后在a上結(jié)合使用列表推導(dǎo)式和fn來只保留同時存在于兩個列表中的值。

map函數(shù)之前已經(jīng)講解過，它會返回一個迭代器，該迭代器會將變換函數(shù)fn應(yīng)用于所有的列表元素上。

set是Python的一個特殊的數(shù)據(jù)類型，是由不重復(fù)元素組成的無序的集。本函數(shù)直接使用set類型消除了列表中的重復(fù)元素。

特別的：當(dāng)過濾條件函數(shù)是lamda x:x時，函數(shù)轉(zhuǎn)化為直接求取兩個列表的交集。

三、列表求差集

實(shí)現(xiàn)三種不同的列表差集的實(shí)現(xiàn)方式。分別是列表直接求差difference，列表中所有元素根據(jù)條件函數(shù)求差difference_by，列表中所有元素根據(jù)條件函數(shù)求對稱差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函數(shù)返回兩個可迭代對象之間的差。該函數(shù)從b中創(chuàng)建一個集合_b，然后在a上使用列表推導(dǎo)式，只保留_b中不包含的值。該函數(shù)中a和b是有順序關(guān)系的，a-b是從a中刪除b中包含的數(shù)據(jù)。

set是Python的一個特殊的數(shù)據(jù)類型，是由不重復(fù)元素組成的無序的集。本函數(shù)直接使用set類型消除了列表中的重復(fù)元素。

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函數(shù)接收兩個列表和一個過濾條件函數(shù)。將提供的函數(shù)應(yīng)用于兩個列表中的每個元素后，返回兩個原始列表的差。函數(shù)通過對b中的每個元素應(yīng)用fn來創(chuàng)建一個集合，然后在a上使用列表推導(dǎo)式與fn相結(jié)合，只保留之前創(chuàng)建的集合_b中沒有包含的值。

特別的：當(dāng)過濾條件函數(shù)是lamda x:x時，函數(shù)轉(zhuǎn)化為直接求取兩個列表的差。

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函數(shù)對兩個列表中的每個列表元素應(yīng)用所提供的函數(shù)后，返回兩個原始列表之間的對稱差。函數(shù)通過分別對每個列表中的每個元素應(yīng)用fn來創(chuàng)建兩個集合_a和_b，然后在每個元素上使用列表理解與fn相結(jié)合，只保留不包含在之前創(chuàng)建的其他集合中的值（在a 中，不在_b中；在b中，不在_a中。）。