Get the nth item of a generator in Python

Question

Is there a more syntactically concise way of writing the following?

gen = (i for i in xrange(10)) index = 5 for i, v in enumerate(gen):     if i is index:         return v 

It seems almost natural that a generator should have a gen[index] expression, that acts as a list, but is functionally identical to the above code.

Answer 1

one method would be to use itertools.islice

>>> gen = (x for x in range(10)) >>> index = 5 >>> next(itertools.islice(gen, index, None)) 5 

Answer 2

You could do this, using count as an example generator:

from itertools import islice, count next(islice(count(), n, n+1))