Anlamanın büyüklüğü nasıl sınırlanır?

Question

list var ve başka bir liste (kavrama yoluyla) oluşturmak istiyorum.

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 çünkü

ile henüz tanımlanmadı: Bu yeni liste

Aşağıdaki kod başarısız olur koşulu aracılığıyla, boyut olarak sınırlı istiyorum Anlama zamanı inşa edilmiştir.

yapı anda yeni listenin boyutunu sınırlamak için bir yolu var mı?

Not: bir karşı ulaşıldığında düzgün break ile for döngüye anlama kırabilir ama anlama kullanan bir mekanizma olup olmadığını bilmek istiyorum. Bu o 3 unsurları üretmek için gerektiğinden daha size daha fazla iş yapmayın sağlar

from itertools import islice 

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

:

Answer 1

Sen yineleme sayısını sınırlamak için, sonra filtreleme yapmak islice() kullanmak için bir jeneratör ifade kullanabilirsiniz .

hiçbir nokta artık burada bir liste anlama kullanarak olmadığını

Not; Liste kavraması kesilemez, sonuna kadar yinelemek üzere kilitlenirsiniz.

Answer 2

@Martijn Pietersitertools.islice bu çözmek için en iyi yol olduğuna tamamen haklı. Ancak, ek bir (harici) kitaplığa sahip değilseniz, bu itertools ve uygulamalarının çoğunu (ve bazı ekleri) içeren iteration_utilities'u kullanabilirsiniz. Bu fonksiyonel programlama gibi en azından, bu biraz daha kolay yapabiliriz: Yalnızca süreç, gerekli görüldüğünde gibi birçok öğeler herhangi arayacak kadar böylece

>>> 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 sınıf içten jeneratörleri kullanan as_* (veya get_*) -Kullanılan.

---

^{Yasal Uyarı: Ben the iteration_utilities library yazarı değilim.}

Answer 3

sayacı istenen tamsayı (3 bu durumda) ulaştığında bir jeneratör için iterating durdurmak için itertools.takewhile bir sayaç oluşturmak için itertools.count kullanıp olabilir:

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

Ya bir yapı inşa benzer bir düşünce Jeneratörü sonlandırmak için StopIteration'u yükseltmek.Yani breaking the list comprehension ait orijinal fikrine alırsınız yakın, ama en iyi uygulama olarak tavsiye etmem:

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

Örnekler:

>>> 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

kullanım numaralandırmak:

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