Python 代碼閱讀合集介紹:為什么不推薦Python初學者直接看項目源碼
本篇閱讀的代碼實現了實現了兩個列表求交集的功能。同時支持使用一個過濾條件函數,列表中所有元素根據該條件求取交集,交集中是兩個列表的原始元素。
本篇閱讀的代碼片段來自於30-seconds-of-python。
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時,函數轉化為直接求取兩個列表的交集。
