《scikit-learn机器学习》 Python代码的埃拉托斯特尼筛法：找出一定范围内所有的素数

埃拉托斯特尼筛法

先用2去筛，即把2留下，把2的倍数剔除掉；再用下一个素数，也就是3筛，把3留下，把3的倍数剔除掉；接下去用下一个素数5筛，把5留下，把5的倍数剔除掉；不断重复下去…

import numpy as np 
a= np.arange(1, 101) 
n_max = int(np.sqrt(len(a))) 
is_prime = np.ones(len(a),dtype=bool) #创建 100 个元素的数组，用来 示记是否为质数
is_prime[0] = False 
for i in range(2,n_max): 
    if i in a[is_prime]: #跳过非质数
        is_prime[(i**2 - 1)::i] = False #减1是为了修复从0始索引的问题
a[is_prime]

def eratosthenes(n):
    IsPrime = [True] * (n + 1)
    for i in range(2, int(n ** 0.5) + 1):
        if IsPrime[i]:
            for j in range(i * i, n + 1, i):
                IsPrime[j] = False
    return [x for x in range(2, n + 1) if IsPrime[x]]
if __name__ == "__main__":
    print(eratosthenes(120))

维基百科