python通過檢查另一個列表中的鍵來獲取元素

Question

我想從列表中獲取子集，方法是檢查其鍵是否在另一個列表中。

class object: 
    def __init__(self, key, x, y): 
     self.key = key 
     self.x = x 
     self.y = y 

o1 = object(key=300, x=18.0, y=100.0) 
o2 = object(key=301, x=12.0, y=1032.0) 

x1 = object(key=100, x=28.0, y=2300.0) 
x2 = object(key=101, x=43.0, y=332.0) 
... 


myobj = [o1, o2, o3] 
objects = [x1, x2, x3, x4, x5, x6, ... ] 

假設o2.key == x3.key和o3.key == x6.key，然後得到[o2, o3]

有沒有一種簡單的方法來做到這一點？

Answer 1

您可以使用以下形式的列表理解：

keys = [ x.key for x in objects ] 
subset = [ x for x in myobj if x.key in keys ] 

雖然你可能會更好調整您的數據轉換成字典，因爲查詢將在更大的數據集有更好的表現。

Answer 2

您可以使用列表理解得到的結果

result = [filter(lambda y: y.key == x.key, myobj) for x in objects] 

編輯：

這將返回列表的列表，這是不是anwser OP需要的，它需要一個額外的reduce到faltten結果

def intersect(): 
    o1 = {"key":300, "x":18.0, "y":100.0} 
    o2 = {"key":500, "x":18.0, "y":100.0} 
    o3 = {"key":600, "x":18.0, "y":100.0} 

    x1 = {"key":300, "x":18.0, "y":100.0} 
    x2 = {"key":300, "x":18.0, "y":100.0} 

    myobj = [o1, o2, o3] 
    objects = [x1, x2] 

    result = reduce(lambda x,y: x+y, [filter(lambda y: y['key'] == x['key'], myobj) for x in objects]) 
    print result 

Answer 3

你可以添加__hash__()和__eq__()方法，對象類型？如果是這樣，你可以直接使用套：

set(myobj).intersection(objects) 

即使你不能使用__hash__()，您可以使用鍵：

subset_keys = set(o.key for o in myobj).intersection(o.key for o in objects) 
subset = [ o for o in objects if o.key in subset_keys ] 

更新：基於更新後的問題...

class object: 
    def __init__(self, key, x, y): 
     self.key = key 
     self.x = x 
     self.y = y 

    def __hash__(self): 
     # must return an int 
     return self.key 
    def __eq__(self, other): 
     return hash(self) == hash(other)