1. 程式人生 > >Numpy攻略:用埃氏篩篩選

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])