求素数

看了半天才看明白。。。对迭代器的理解还是不够啊。思路不够开阔。

 

# 迭代器对象从3开始每次加2迭代
def _odd_iter():
    n = 1
    while True:
        n = n+2
        yield n

# 返回匿名函数
def _not_divisible(n):
    return lambda x:x%n>0

# 这也是一个迭代器函数
def primes():
    # 第一次返回2
    yield 2
    # 初始化迭代器函数对象
    it = _odd_iter()
    while True:
        # 每次迭代获取下一个对象
        n = next(it)
        # 返回这个对象
        yield n
        # 移除迭代器中和本次所取数据求模等于0，及可以被该数据整除的对象（这里迭代数据成为了匿名函数的参数x）
        it = filter(_not_divisible(n),it)

for n in primes():
    if n <1000:
        print(n)
    else:
        break

这个算法的重点在于对迭代器的使用。利用惰性计算的序列实现简洁的算法。