素数生成のコードがうまく動かない

Question

実現したいこと

pythonで素数を生成するジェネレータを作りたいです
エラトステネスの篩と無限イテレータを使っています
素数を生成するはずが生成された数列は2, 3, 4となってしまいます
エラーは発生していません

該当のソースコード

# coding: utf-8

import itertools

def gen_prime():
    primes = itertools.count(2)
    
    while True:
        p = next(primes)
        primes = itertools.ifilter(lambda x: x % p != 0, primes)
        yield p

p = gen_prime()
for _ in range(3):
    print(next(p)) # 2, 3, 4

試したこと

primes = itertools.ifilter(lambda x: x % p != 0, primes)
primes, tmp = itertools.tee(primes)
list(itertools.islice(tmp, 10))

とすると最初の10個は素数になるが27(合成数)が出てくるようになる

Answer 1

python2の環境が手元にないから試せませんが、この問題じゃないでしょうか。

ループの中で異なる値で定義されたラムダ式が、同じ値を返すのはなぜですか？

Comments

[room]

ありがとうございます

def mulcheck(p):
return lambda x: x % p != 0
みたいにしたら上手く行きました

[LouiS0616]

primes = itertools.ifilter(lambda x, p=p: x % p != 0, primes) みたいに p=p を加えるだけで大丈夫そうですよ。

[room]

ほんとだ
わざわざありがとうございます

[TakaiY]

勉強になりました。

Answer 2

イテレータを使わずに実装した方が簡単です。

def gen_prime():
    def check(n, primes):
        for p in primes:
            if n % p == 0:
                return False
        return True
    yield 2
    n = 3
    primes = [2]
    while True:
        if check(n, primes):
            yield n
            primes.append(n)
        else:
            pass
        n += 2

p = gen_prime()
for i in range(10):
    print(next(p))

実行結果

>>> for i in range(10):
...     print(next(p))
...
2
3
5
7
11
13
17
19
23
29

Comments

[room]

かなり大きい値まで使うので計算量的にエラトステネスの篩が使いたかったのとジェネーレータにしたかったので無限イテレータを使いたかったんです

回答ありがとうございました