如何限制理解的大小？

Question

我有一個list並想構建（通過理解）另一個列表。我想在大小受到限制，這個新的列表，通過條件

下面的代碼將失敗：

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

與

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 

因爲b在沒有定義尚未理解建立的時間。

有沒有辦法在構建時限制新列表的大小？

注：我能打破理解成for循環用正確的break到達櫃檯時，但我想知道是否有一種使用一個理解的機制。

Answer 1

您可以用生成器表達式做過濾，然後用islice()限制迭代次數：

from itertools import islice 

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

這保證比你要生產這些3個元素你沒有做更多的工作。

請注意，在這裏使用列表理解沒有任何意義;列表理解不能被打破，你被鎖定在迭代到最後。

Answer 2

@Martijn Pieters完全正確，itertools.islice是解決此問題的最佳方法。但是，如果您不介意附加（外部）庫，則可以使用iteration_utilities，其中包含許多itertools及其應用程序（以及其他一些應用程序）。它可以使這個更容易一點，至少，如果你喜歡函數式編程：

>>> from iteration_utilities import Iterable 

>>> Iterable([1, 2, 1, 2, 1, 2]).filter((1).__eq__)[:2].as_list() 
[1, 1] 

>>> (Iterable([1, 2, 1, 2, 1, 2]) 
...   .filter((1).__eq__) # like "if item == 1" 
...   [:2]     # like "islice(iterable, 2)" 
...   .as_list())   # like "list(iterable)" 
[1, 1] 

的iteration_utilities.Iterable類使用發電機內部所以它只能處理儘可能多的項目neccessary，直到你調用任何as_*（或get_*） -方法。

---

^{免責聲明：我的the iteration_utilities library作者。}

Answer 3

你可以使用itertools.count生成一個計數器和itertools.takewhile在發電機停止迭代當計數器達到（在這種情況下3）所需的整數

from itertools import count, takewhile 
c = count() 
b = list(takewhile(lambda x: next(c) < 3, (i for i in a if i == 1))) 

或類似的想法建立一個結構提高StopIteration來終止發電機。這就是你會得到你原來的想法breaking the list comprehension最接近的，但我不會推薦它作爲最佳實踐：

c = count() 
b = list(i if next(c) < 3 else next(iter([])) for i in a if i == 1) 

例子：

>>> a = [1,2,1,4,1,1,1,1] 

>>> c = count() 
>>> list(takewhile(lambda x: next(c) < 3, (i for i in a if i == 1))) 
[1, 1, 1] 

>>> c = count() 
>>> list(i if next(c) < 3 else next(iter([])) for i in a if i == 1) 
[1, 1, 1] 

Answer 4

使用枚舉：

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