- from itertools import imap, starmap
- def gen():
- for i in range(10):
- print "generating %s" % i
- yield (i, i*2)
- def mapper(item, item_squared):
- print "mapping %s" % (item, )
- return "mapped %s" % (item)
- starmap_gen = starmap(mapper, gen())
- imap_gen = imap(mapper, gen())
- print list(starmap_gen)
- print "--"
- print list(imap_gen)