python设置与对象集的交集

Question

我正在使用 amazon boto，我有 2 个列表。列表 1 包含 Instance 对象。清单 2 包含 InstanceInfo 对象。这两个对象都有一个名为 id 的属性。我需要获取 InstanceInfo 列表中存在 id 的 Instance 对象列表。

l1 = [Instance:i-04072534, Instance:i-06072536, Instance:i-08072538, Instance:i-0a07253a, Instance:i-e68fa1d6, Instance:i-e88fa1d8, Instance:i-ea8fa1da, Instance:i-ec8fa1dc]

l2 = [InstanceInfo:i-ec8fa1dc, InstanceInfo:i-ea8fa1da, InstanceInfo:i-e88fa1d8, InstanceInfo:i-e68fa1d6]

想要的结果：

l3 = [Instance:i-ec8fa1dc, Instance:i-ea8fa1da, Instance:i-e88fa1d8, Instance:i-e68fa1d6]

现在我已经解决了：

l3= []
for a in l1  
    for b in l2:
        if a.id == b.id:
            l3.append(a)

但是，有人告诉我，我应该使用设置交集来替换它。我一直在看例子，它看起来非常简单。但是我没有看到任何使用对象的示例。

我已经玩了一段时间，理论上我可以看到它工作，但可能有一些我可能不知道的“高级”语法。我还在学习python。

Answer 1

这比 Marcin 的回答更快（虽然相似）：

ids_l1 = set(x.id for x in l1)  # All ids in list 1
intersection = [item for item in l2 if item.id in ids_l1]  # Only those elements of l2 with an id in l1

预先计算 ids_l1 并且不要写 if item.id in set(…) 很重要，因为每次都会重新构建集合（因为要为每个元素 item 重新评估完整的测试表达式）。

Python 集为您提供快速的元素成员资格测试 (in)。这样的测试用集合比用列表快得多（因为列表的元素必须一个一个地读取，而集合的元素是“散列的”）。

Answer 2

您的方法对于小型列表可能相对有效。

使用集合，您必须提取 id，计算 id 的交集，然后将项目收集到新列表中。比如：

set1 = set(x.id for x in l1)
set2 = set(x.id for x in l2)
intersection_ids = set1 & set2
intersection_list = [item for item in l2 if item.id in intersection_ids]

您可以通过扫描较短的列表或将对象存储在字典中来提高效率。

Answer 3

试试这个：

# get ids of elements in second list
l2_ids = [x.id for x in l2]
# get elements from first list that have ids in second
l3 = [x for x in l1 if x.id in l2_ids]