1. 程式人生 > 其它 >Python程式碼閱讀(第16篇):列表求差集

Python程式碼閱讀(第16篇):列表求差集

本篇閱讀三種不同的列表差集的實現方式。分別是列表直接求差difference,列表中所有元素根據條件函式求差difference_by,列表中所有元素根據條件函式求對稱差symmetric_difference_by。 本篇閱讀的程式碼片段來自於30-seconds-of-python。

Python 程式碼閱讀合集介紹:為什麼不推薦Python初學者直接看專案原始碼

本篇閱讀三種不同的列表差集的實現方式。分別是列表直接求差difference,列表中所有元素根據條件函式求差difference_by,列表中所有元素根據條件函式求對稱差symmetric_difference_by

本篇閱讀的程式碼片段來自於30-seconds-of-python

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中不包含的值。該函式中ab是有順序關係的,a-b是從a中刪除b中包含的資料。

setPython的一個特殊的資料型別,是由不重複元素組成的無序的集。本函式直接使用set型別消除了列表中的重複元素。

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時,函式轉化為直接求取兩個列表的差。

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時,函式轉化為直接求取兩個列表的對稱差。