When iterators behave as lists

Some list operations also work on iterators. We will now examine the equivalents of list comprehensions and list zipping (refer to section List of Chapter 3, Container Types, for more details).

Generator expression

There is an equivalent of list comprehension for generators. Such a construction is called a generator expression:

g = (n for n in range(1000) if not n % 100)
# generator for  100, 200, ... , 900

This is useful in particular for computing sums or products because those operations are incremental; they only need one element at a time:

sum(n for n in range(1000) if not n % 100) # returns 4500 

In that code, you notice that the sum function is given one argument, which is a generator expression. Note that Python ...

Get Scientific Computing with Python 3 now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.