python - 选择满足条件的元素

Question

我想从一个(很长的)列表中选择一个满足条件的特定元素。在我的具体情况下，列表中的每个项目都是一个类，我想要的条件是，例如，item.id == 1。

我开始了

[item for item in collection if item.id == 1]

但我不想遍历整个列表，因为我确定只有一个元素满足条件。

另一种选择是

def check(collection):
    for item in collection:
        if item.id == 1:
            return item

有没有更有效的方法？ Python 3 有内置函数吗？另一种选择是预处理列表以获得字典，例如

{item.id : item for item in collection}

Answer 1

您可以使用 generator expression和 next() function :

item = next((item for item in collection if item.id == 1), None)

这将在集合中找到第一个 匹配元素，如果没有匹配则返回None。它不会迭代和评估任何超过找到一个元素的严格要求。

但是，如果您确实需要做很多这样的事情，那么只创建一次映射然后每次都查询会更有效率:

mapping = {item.id: item for item in collection}
item = mapping.get(1)

因为这会将循环限制为一次，然后使用 O(1) 查找性能。这是内存与速度的权衡。