我想从一个(很长的)列表中选择一个满足条件的特定元素。在我的具体情况下,列表中的每个项目都是一个类,我想要的条件是,例如,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}
请您参考如下方法:
您可以使用 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) 查找性能。这是内存与速度的权衡。
