Python代碼閱讀(第15篇):列表求交集


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應用於所有的列表元素上。

setPython的一個特殊的數據類型,是由不重復元素組成的無序的集。本函數直接使用set類型消除了列表中的重復元素。

特別的,當過濾條件函數是lamda x:x時,函數轉化為直接求取兩個列表的交集。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM