python：如何獲得列表生成器表達式中的列表長度？

Question

也許像這樣（只是一個僞代碼）

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

什麼是Python的方式做那件事？

Answer 1

您可以使用itertools.count()來跟蹤你有多少元素使用：

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 

用於跟蹤添加到列表中的元素，列表的，因此它的長度，得出：

[1, 2] 

如預期與長度

2.

EDITED：更簡單的方法，用enumerate：

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