题目7：找出第10001个质数

张无忌

1. # -*- coding: utf-8 -*-
   
2. """
   
3. 10001st prime number.
   
4. """
   
5. from math import sqrt, floor
   
6. 
   
7. def is_prime(number):
   
8.     if number > 1:
   
9.         if number == 2:
   
10.             return True
    
11.         if number%2 == 0:
    
12.             return False
    
13.         for i in range(3, floor(sqrt(number)) + 1, 2):
    
14.             if number%i == 0:
    
15.                 return False
    
16.         return True
    
17.     return False
    
18. 
    
19. def find_prime():
    
20.     n = 2
    
21.     while True:
    
22.         if is_prime(n):
    
23.             yield n
    
24.         n += 1
    
25. 
    
26. fp = find_prime()
    
27. 
    
28. i, max_n = 1, 10001
    
29. while i < max_n:
    
30.     next(fp)
    
31.     i += 1
    
32. 
    
33. print(next(fp))

复制代码