예제 #1
0
from problem0003 import primes
from itertools import islice


def nth(iterable, n):
    '''return n-th element from given iterable
    >>> nth([1, 2, 3, 4], 2)
    2
    '''
    return next(islice(iterable, n-1, None))

if __name__ == '__main__':
    print(nth(primes(), 10001))