Python中的简单Prime生成器

def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, int(n**0.5)+1):
        if n % i == 0:
            return False
    return True

def prime_generator(n):
    primes = []
    for i in range(2, n):
        if is_prime(i):
            primes.append(i)
    return primes