Python中的懶惰地圖函數

Question

有沒有辦法讓map懶惰？還是有Python的內置的另一個實現？

我想是這樣工作的：

from itertools import count 

for x in map(lambda x: x**2, count()): 
    print x 

當然，上面的代碼不會結束，但我想剛進入任何條件（或更復雜的邏輯）的for內，在某個點停下來。

Answer 1

上的Python 2.x的使用itertools.imap或升級到Python 3.x的

你也可以使用一個簡單的生成器表達式是遠遠更Python：

foo = (x**2 for x in count()) 

Answer 2

itetools.imap是懶惰的。

In [3]: itertools.imap? 
Type:  type 
String Form:<type 'itertools.imap'> 
Docstring: 
imap(func, *iterables) --> imap object 

Make an iterator that computes the function using arguments from 
each of the iterables. Like map() except that it returns 
an iterator instead of a list and that it stops when the shortest 
iterable is exhausted instead of filling in None for shorter 
iterables.