python通过检查另一个lis中的键来获取元素

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

你能给对象类型添加^{}和__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)