Python 获取和判断质数

zltzlt

1. def is_prime(num: int) -> bool:
   
2.     """
   
3.     判断一个数是不是质数
   
4. 
   
5.     参数
   
6.     ----------
   
7.     num：需要判断的质数
   
8.     """
   
9.     if num == 2:
   
10.         return False
    
11.     for a in range(2, num // 2 + 1):
    
12.         if num % a == 0:
    
13.             return False
    
14.     return True
    
15. 
    
16. 
    
17. def get_prime(start: int, stop: int) -> list:
    
18.     """
    
19.     获取从 start 到 stop 的质数
    
20. 
    
21.     参数
    
22.     ----------
    
23.     start：开始的范围
    
24.     stop：停止的范围（不包括）
    
25.     """
    
26.     assert start >= 2
    
27.     return list(filter(is_prime, range(start, stop)))

复制代码

永恒的蓝色梦想

更加高效的版本

1. def isPrime(num):
   
2.         num=abs(num)
   
3.         if num<=2 or not num&1:
   
4.                 return False
   
5.         for i in range(3,int(num**0.5),2):
   
6.                 if not num%i:
   
7.                         return False
   
8.         return True

复制代码