Python：基于生成器和filter实现惰性求值 - 以筛法质数计算为例

函数式编程值得称道的是惰性计算概念的引入，以Haskell实现埃氏筛法计算质数的代码作为代表：

primes = filterPrime [2..]
  where filterPrime (p:xs) =
          p : filterPrime [x | x <- xs, x `mod` p /= 0]

在Python中通过生成器实现类似的功能。首先我们实现一个从2开始生成自然数的generator来对标[2..]：

def count_from_2():
    return itertools.count(start=2)

然后定义一个筛选函数作为filter的参数，对标x mod p /= 0. 这里我们用一个高阶函数来定义，作为模板：

def _not_divisible_by(n):
    return lambda x: x % n > 0

接下来照搬定义filterPrime和primes

def filter_prime(nums):
    while True:
        p = next(nums)
        yield p
        nums = filter(_not_divisible_by(p), nums)

def primes():
    return filter_prime(count_from_2())

进行测试：

if __name__ == '__main__':
    for i in itertools.takewhile(lambda n: n < 100, primes()):
        print(i, end=' ')
    print()

% python main.py
2 3 5 7 11 13 17 19 23 29 31 37 41 43 47 53 59 61 67 71 73 79 83 89 97 

完整代码：

import itertools

def count_from_2():
    return itertools.count(start=2)

def _not_divisible_by(n):
    return lambda x: x % n > 0

def filter_prime(nums):
    while True:
        p = next(nums)
        yield p
        nums = filter(_not_divisible_by(p), nums)

def primes():
    return filter_prime(count_from_2())

if __name__ == '__main__':
    for i in itertools.takewhile(lambda n: n < 100, primes()):
        print(i, end=' ')
    print()