用filter()求素数

# 这个简单，但有一个问题就是2出不来
import math
import sys

def prime(n):# n的取值2，3，4，5。。。。
    if n <= 1:
        return 0
    for i in range(2,n):
        if n % i == 0:#相当于3%2；4%2，4%3；5%2，5%3，5%4 。。。。。
            return 0
    return 1

if __name__ == "__main__":
    n = int(sys.argv[1])
    for i in range(2,n+1):
        if prime(i):
            print i



# 先构造一个从3开始的奇数序列，注意这是一个生成器，并且是一个无限序列。
def _odd_iter():
    n = 1
    while True:
        n = n + 2
        yield n

# 定义一个筛选函数       
def _not_divisible(n):
    return lambda x: x % n > 0 # 这里很重要，我的理解就是第一个n=3?然后所有自然数x%3>0成立的都保留（意思就是3的倍数全删了),然后依次往下
                               # 可是2的倍数怎么弄？


# 最后定义一个生成器，不断返回下一个素数。
# 先返回第一个素数2，然后利用filter()不断产生筛选后的新的序列。
def primes():
    yield 2
    it = _odd_iter() # 初始序列
    while True:
        n = next(it) # 返回序列的第一个数
        yield n
        it = filter(_not_divisible(n), it) # 构造新序列

# 由于primes（）是一个无限序列，所以调用时需要设置一个退出循环的条件。         
for n in primes():
    if n < 1000:
        print(n)
    else:
        break