如何限制推导式的大小？

我有一个list并想要构建（通过理解）另一个列表。我希望通过一个条件限制这个新列表的大小

下面的代码将会失败：

a = [1, 2, 1, 2, 1, 2]
b = [i for i in a if i == 1 and len(b) < 3]

with

Traceback (most recent call last):
  File "compr.py", line 2, in <module>
    b = [i for i in a if i == 1 and len(b) < 3]
  File "compr.py", line 2, in <listcomp>
    b = [i for i in a if i == 1 and len(b) < 3]
NameError: name 'b' is not defined

because b没有定义yet当理解力建立起来时。

有没有办法在构建时限制新列表的大小？

注意：我可以将理解分解为for用适当的循环break当达到计数器时，但我想知道是否有一种使用理解的机制。

您可以使用生成器表达式进行过滤，然后使用islice()限制迭代次数：

from itertools import islice

filtered = (i for i in a if i == 1)
b = list(islice(filtered, 3))

这可以确保您不会做超出生产这 3 个元素所需的工作。

请注意，这里不再使用列表理解了；列表理解无法被打破，你被锁定迭代到最后。