python中使用生成器的素数答案

【问题标题】：Prime numbers in python using generatorspython中使用生成器的素数
【发布时间】：2017-06-20 17:03:32
【问题描述】：

我正在尝试在 python 中编写一个程序，以使用类似于在 Haskell 中编写素数程序（在他们的网站顶部找到的程序）中使用的逻辑来生成素数。我写了以下两个生成器函数。

def integers():
    i=2
    while True:
        yield i
        i+=1

def primes(li):
    x=li
    while True:
        i=next(x)
        y=(j for j in x if %i!=0)
        yield i
        x=y

整数是数字 2,3,4,5... 的生成器所以我的猜测是

a=primes(integers())

应该给出一个素数的生成器，但它什么也没给出。这是我得到的输出。

>>> next(a)
2
>>> next(a)
3
>>> next(a)
4
>>> next(a)
5
>>> next(a)
6

有什么建议吗？

【问题讨论】：

Prime generator in python的可能重复
为什么你的 integers 函数会生成 2 的倍数？您可以通过在每次迭代中将 i 增加 2 而不是 1 来避免这种情况。
@ForceBru 我相信其目的是重现页面顶部的实现here

标签： python

【解决方案1】：

好吧，你不会在 Python 中使用这种方法，因为你通常会避免递归。但这里有一个有点等价的实现，在itertools 的帮助下：

>>> from itertools import count, chain
>>> def filter_primes(integers):
...     p = next(integers)
...     yield from chain([p], filter_primes(x for x in integers if x%p !=0))
...
>>> primes = filter_primes(count(2))
>>> next(primes)
2
>>> next(primes)
3
>>> next(primes)
5
>>> next(primes)
7
>>> next(primes)
11
>>> next(primes)
13
>>> next(primes)
17
>>> next(primes)
19
>>> next(primes)
23
>>> next(primes)
29

使用更多itertools帮助获取前20名：

>>> primes = filter_primes(count(2))
>>> list(islice(primes, 20))
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71]
>>>

【讨论】：

【解决方案2】：

您的代码看起来很合理。它不起作用，因为它做了一个令人惊讶的错误假设：生成器表达式关闭了它使用的值。

相反，(j for j in x if j % i != 0) 不会关闭 i！这就是为什么在稍后执行过程中更改i 会更改过滤表达式。不是针对连续的素数构建一堆过滤器，而是整个生成器链在每一步都发生变异以过滤最后一个产生的价值。

演示：

>>> v = 1
>>> gen = (x for x in itertools.count() if x % v == 0)
>>> print next(gen), next(gen), next(gen)
0 1 2
>>> v = 3
>>> print next(gen), next(gen), next(gen)
3 6 9

递归解决方案是解决问题的明显方法。

另一种方法是使用 lambda 来执行闭包（参见 IIFE）：

def primes(source):
    while True:
        value = next(source)
        yield value
        source = (lambda v: (j for j in source if j % v != 0))(value)

这行得通，但它产生的开销与直接递归大致相同，而且读起来更糟。不过，它不会花费调用堆栈。

【讨论】：