Numpy攻略:用埃氏筛筛选

埃氏筛:筛选质数的一种算法,用迭代的方式识别出已经找到的质数的倍数,能高效地筛选出小于一千万的质数。让我们去试着寻找10001个质数。

具体步骤如下:
1.创建一个连续的整数列表:用arange函数
2.筛选出p的倍数
完整代码如下图:

 import numpy
LIM=10**6
N=10**9
P=10001
primes=[]
p=2
#通过列出前6个质数:2,3,5,7,11,13,我们看到第6个质数是13
#第10001个质数是多少
def check_primes(a,p):
    #2.筛选出p的倍数
    a=a[a%p!=0]
    return a
for i in range(3,N,LIM):
    #1.创建一个整数列表
    a=numpy.arange(i,i+LIM,2)
    while len(primes)<p:
        a=check_primes(a,p)
        primes.append(p)
        p=a[0]
print(len(primes),primes[p-1])

猜你喜欢

转载自blog.csdn.net/JiangCaifu/article/details/83898924
今日推荐