【视频49课后作业】求2000000以内的素数之和

payton24

这节主要讲生成器，如果抛开一定要用yield语句来生成generator的话。
感觉自建元组的生成器更容易理解，运行效率也差不多。

1. ##生成2000000以内的循环数，step=1,start=0
   
2. ##提取出素数
   
3. ##判断循环数是否素数
   
4. ##如果是素数，加入到sum里面。
   
5. ##def get_prime():
   
6.    
   
7. import math
   
8. def is_prime(number):
   
9.     if number > 1:
   
10.         if number == 2:
    
11.             return True
    
12.         if number % 2 == 0:
    
13.             return False
    
14.         for current in range(3, int(math.sqrt(number) + 1), 2):
    
15.             if number % current == 0:
    
16.                 return False
    
17.         return True
    
18.     return False
    
19.    
    
20. def main():
    
21.     sum=0
    
22.     nums = (each for each in range(2000000))    ##生成器
    
23.     for i in nums:
    
24.         if is_prime(i):
    
25.             sum +=i
    
26.     print(sum)
    
27. 
    
28. if __name__ == '__main__':
    
29.     main()
    

复制代码

jerryxjr1220

range函数本身就是生成器，何必多此一举？

payton24

jerryxjr1220 发表于 2017-12-13 09:36
range函数本身就是生成器，何必多此一举？

谢谢这位大牛，初学者嘛，让您见笑了。您的意思是：

1.     nums = (each for each in range(2000000))    ##生成器
   
2.     for i in nums:

复制代码

与下面语句是等效率的吗？

1.     nums =range(2000000)
   
2.     for i in nums:

复制代码

payton24

jerryxjr1220 发表于 2017-12-13 09:36
range函数本身就是生成器，何必多此一举？

另外，我用简单程序测算了一下，使用生成器和一般的for循环的运行时间都是差不多的。能指点一下吗？
代码一如下：

1. import time
   
2. import sys
   
3. 
   
4. time_start = time.time()
   
5. 
   
6. '''------------------------------------------------------------'''   
   
7. import math
   
8. def is_prime(number):
   
9.     if number > 1:
   
10.         if number == 2:
    
11.             return True
    
12.         if number % 2 == 0:
    
13.             return False
    
14.         for current in range(3, int(math.sqrt(number) + 1), 2):
    
15.             if number % current == 0:
    
16.                 return False
    
17.         return True
    
18.     return False
    
19.    
    
20. def main():
    
21.     sum=0
    
22. ##    nums = (each for each in range(2000000))
    
23.     nums =range(2000000)
    
24.     for i in nums:
    
25.         if is_prime(i):
    
26.             sum +=i
    
27.     print(sum)
    
28. 
    
29. main()
    
30. '''------------------------------------------------------------'''
    
31. 
    
32. time_end = time.time()
    
33. print('返回结果花费的时间： %s' % (time_end - time_start))
    
34. print('返回结果占用内存大小：%s' % sys.getsizeof(main()))

复制代码

IDLE结果为：
======================== RESTART: F:\Python-Ex\aa.py ========================
142913828922
返回结果花费的时间： 9.932051420211792
142913828922
返回结果占用内存大小：16
>>>

代码二如下：

1. import time
   
2. import sys
   
3. 
   
4. time_start = time.time()
   
5. 
   
6. '''------------------------------------------------------------'''
   
7. import math
   
8. def is_prime(number):
   
9. ##    print('num %d' %number)
   
10.     if number > 1:
    
11.         if number == 2:
    
12.             return True
    
13.         if number % 2 == 0:
    
14.             return False
    
15.         for current in range(3, int(math.sqrt(number) + 1), 2):
    
16.             if number % current == 0:
    
17.                 return False
    
18.         return True
    
19.     return False
    
20. 
    
21. def get_primes(number):
    
22.     while True:
    
23.         if is_prime(number):
    
24.             yield number
    
25.         number +=1
    
26.    
    
27. def solve():
    
28.     sum=2
    
29.     for next_prime in get_primes(3):
    
30.         if next_prime < 2000000:
    
31.             sum +=next_prime
    
32.         else:
    
33.             print(sum)
    
34.             return
    
35. 
    
36. if __name__ == '__main__':
    
37.     solve()
    
38. '''------------------------------------------------------------'''
    
39. 
    
40. time_end = time.time()
    
41. print('返回结果花费的时间： %s' % (time_end - time_start))
    
42. print('返回结果占用内存大小：%s' % sys.getsizeof(solve()))

复制代码

IDLE结果为：
======================== RESTART: F:\Python-Ex\aa.py ========================
142913828922
返回结果花费的时间： 9.714050531387329
142913828922
返回结果占用内存大小：16
>>>

jerryxjr1220

payton24 发表于 2017-12-13 11:05
另外，我用简单程序测算了一下，使用生成器和一般的for循环的运行时间都是差不多的。能指点一下吗？
...

你真的想了解range的底层逻辑的话，可以去看range的底层代码，看是如何实现的。
如果你只是初学者，可以先不必深究这个问题，只要知道range函数本身也是调用生成器产生的，所以不必要用生成器再产生生成器。生成器与生成器在代码逻辑相同的情况下，运行效率的差别是很小的。

payton24

jerryxjr1220 发表于 2017-12-13 15:05
你真的想了解range的底层逻辑的话，可以去看range的底层代码，看是如何实现的。
如果你只是初学者，可以 ...

多些指点，明白了。希望以后能看得懂底层逻辑，现阶段先记着。