在搜索Python的素数生成器时,我发现:
def primes(n):
if n==2: return [2]
elif n<2: return []
s=range(3,n+1,2)
mroot = n ** 0.5
half=(n+1)/2-1
i=0
m=3
while m <= mroot:
if s[i]:
j=(m*m-3)/2
s[j]=0
while j<half:
s[j]=0
j+=m
i=i+1
m=2*i+3
return [2]+[x for x in s if x]
print primes(13)
print primes(3000)
并且:
def get_primes(number):
while True:
if is_prime(number): #is_prime is defined somewhere else
yield number
number += 1
什么更有效,返回列表或yield命令?为什么?考虑我正在寻找大量的质数,例如前1000个素数。顺便说一下,第二段代码似乎是一个无限循环,如何停止它?
谢谢,并抱歉问了这么多问题。