目的:使用埃氏筛法构造素数
计算素数的一个方法是埃氏筛法,它的算法理解起来非常简单:
首先,列出从2开始的所有自然数,构造一个序列:
2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取序列的第一个数2,它一定是素数,然后用2把序列的2的倍数筛掉:
3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取新序列的第一个数3,它一定是素数,然后用3把序列的3的倍数筛掉:
5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取新序列的第一个数5,然后用5把序列的5的倍数筛掉:
7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
不断筛下去,就可以得到所有的素数。
代码实现
#_*_coding:utf_8_*_
def iters():#先构造一个从3开始的奇数序列。这是一个生成器,并且是一个无限序列
n=1
while 1:
n=n+2
yield n
def isinit(n):#筛选函数
return lambda x:x%n>0
def prime():
yield 2
it=iters()# 初始序列
while 1:
n=next(it)# 返回序列的第一个数
yield n
it=filter(isinit(n),it)# 构造新序列
for n in prime():
if(n<10):
print(n)
else:
break
问题
在python3中能够直接运行,完美出结果,但是在python2中只出现第一个结果,后面就卡死了。原因是什么呢?搜了好多资料没有这个的说明,在大神那里得到了答案。
filter() 函数:python2 会把你的 it 遍历完再直接返回一个 list。python3 不需要遍历完,它返回一个iterable
解决方法:
自己实现filter方法,重写里面的逻辑,如下:
def filter(func, it):
while 1:
v = next(it)
if func(v):
yield v
至此,上面代码在python2中也可以完美运行了。