Python: Wie bekomme ich die Listenlänge im Ausdruck des Listengenerators?

Question

Vielleicht wie (nur ein Pseudo-Code)

li = [1,2,3,4,5,6] 
a = [d for d in li if self.length<3] 
print a 
[1,2] 

Was für einen pythonic Weg ist, etwas zu tun?

Answer 1

Sie itertools.count() verwenden können, um zu verfolgen, wie viele Elemente, die Sie verwendet haben:

from itertools import count 

li = [1,2,3,4,5,6] 
c = count(1) # initialize to 1 
a = [d for d in li if next(c) < 3] # everytime you add an element 
print (a)       # in the comprehension `c` is incremented 

, die auf der Liste hinzugefügt Spur der Elemente hält, damit die Länge der Liste, die gibt:

[1, 2] 

wie mit einer Länge erwartet 2.

EDITED: einfacher Art und Weise, mit enumerate:

a = [d for idx, d in enumerate(li) if idx < 2]