在兩個列表中查找對應關係的最快方法Python

Question

假設我有一個具有屬性from_id的對象的列表lst1。另外，我還有另一個具有屬性to_id的對象的列表lst2。

我需要爲lst1中的每個對象找到它的「對」lst2（from_id應該等於to_id）。

lst2中最多隻能有一個對象（零個或一個），對應於lst1中的給定對象。

什麼是在Python中實現它的最快方法？

Answer 1

使用套;因爲它像一個沒有鍵的字典（只是值）。首先將您的列表轉換爲集合。

lst1 = set(lst1) 
lst2 = set(lst2) 

res = [(obj, to_obj) for obj in lst1 for to_obj in lst2 if obj.from_id == to_obj.to_id] 

這可以被打破

result = [] 
for obj in lst1: 
    for to_obj in lst2: 
     if obj.from_id == to_obj.to_id: 
      result.append((obj, to_obj)) 
      break 

這將元組的列表結束。 較長的版本會破壞找到該對象的內容，但理解代碼不會。