Python primes

primes

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import itertools

def gen(i):
    while True:
        yield i
        i = i + 1

def primes(gen):
    p = next(gen)
    yield p
    
    def g():
        for i in gen:
            if i % p > 0:
                yield i

    for i in primes(g()):
        yield i

def prime(x):
    return next(itertools.islice(primes(gen(2)), x, x+1))