Python笔记16-numpy求质数

最新推荐文章于 2024-03-04 14:24:46 发布

q275343119

最新推荐文章于 2024-03-04 14:24:46 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/q275343119/article/details/89504644

本文介绍了一种利用Python的NumPy库高效求解指定范围内素数的方法。通过创建逻辑数组并采用筛法原理，该算法能够快速确定哪些数字是素数，显著减少了计算时间和资源消耗。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
numbers = np.arange(1,100)	# 假设求1-100
length = len(numbers)	# 求数组长度，其实也是求最后一个数
n_max = int(np.sqrt(length))	# 对长度开根号，减少遍历长度
is_prime = np.ones(length,dtype = bool)	# 建立一个和数字数组等长的逻辑数组
is_prime[0] = False	# 逻辑数组第一个值为False 遍历的时候排除1

for i in range(2,n_max):	# 从2开始遍历
    if i in numbers[is_prime]:	# 判断这个i在不在素数数组里面
        is_prime[(i**2-1)::i] = False	# 如果在，则从i**2-1 开始以i为步长取得的数肯定不是素数

print(numbers[is_prime])