使用列表推导在列表中查找 1 个元素？

Question

我有一个关于 Python 最佳实践的简短问题。我已经看到 lambda + filter 函数在列表理解中的使用，但我想知道是否有更简单的方法可以在列表中查找特定元素，而无需生成和迭代整个列表。

使用 AWS boto3 库，我正在使用列表推导查询各种 get_x() 函数：

[i['domainName'] for i in domain_names['items'] if re.search(r'\b'+domain_name, i['domainName'])].pop()

[i['id'] for i in usage_plans['items'] if i['name']==f'{self.service}Usage'].pop()

如果未找到项目，则会捕获 IndexError 并将其转发给用户。由于这是在 AWS Lambda 函数中，我担心函数的可扩展性和运行时计费。

我应该继续使用列表推导式，还是有更好的方法？

Answer 1

如果您想避免遍历整个列表，您可以使用生成器推导而不是列表推导。例如：

next(i for i in range(0, 2**1000) if i % 2 == 1)

遍历整个范围需要一段时间，但使用生成器理解它是瞬时的。

请注意，如果它没有找到一个项目，您会得到一个StopIteration 异常而不是IndexError。您确实必须捕获它并将其包装在另一个异常中，因为通过堆栈传播的流浪 StopIteration 可能会导致奇怪的行为。

包装StopIteration 看起来像这样：

>>> try:
...   next(i for i in range(0, 100) if i % 2 == 3 )
... except StopIteration:
...   raise IndexError("Couldn't find item")
... 
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
StopIteration

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
IndexError: Couldn't find item

请注意，您可以给next 一个默认值以返回而不是提高StopIteration：

>>> print(next((i for i in range(0, 100) if i % 2 == 3), None))
None