Chức năng bản đồ lười biếng trong Python

Question

Có cách nào để làm cho map lười biếng không? Hoặc là có một thực hiện khác của nó được xây dựng trong Python?

Tôi muốn một cái gì đó như thế này để làm việc:

from itertools import count 

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

Tất nhiên, các mã trên sẽ không kết thúc, nhưng tôi muốn chỉ để nhập bất kỳ điều kiện (hoặc logic phức tạp hơn) bên trong for và dừng lại ở một số điểm.

Answer 1

sử dụng itertools.imap trên Python 2.x hoặc nâng cấp lên Python 3.x

Bạn cũng có thể chỉ cần sử dụng một biểu thức máy phát điện đơn giản mà là xa pythonic hơn:

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

Answer 2

itetools.imap lười.

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.