赞
踩
filter()
函数用于过滤序列。
和map()
类似,filter()
也接收一个函数和一个序列。和map()
不同的是,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。
例如,在一个list中,删掉偶数,只保留奇数,可以这么写:
- def is_odd(n):
- return n % 2 == 1
-
- list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
- # 结果: [1, 5, 9, 15]
可见用filter()
这个高阶函数,关键在于正确实现一个“筛选”函数。
注意到filter()
函数返回的是一个Iterator
,也就是一个惰性序列,所以要强迫filter()
完成计算结果,需要用list()
函数获得所有结果并返回list。
计算素数的一个方法是埃氏筛法,它的算法理解起来非常简单:
首先,列出从2
开始的所有自然数,构造一个序列:
2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
取序列的第一个数2
,它一定是素数,然后用2
把序列的2
的倍数筛掉:
3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
取新序列的第一个数3
,它一定是素数,然后用3
把序列的3
的倍数筛掉:
5, 6, 7, 8, 9, 10, 11, 12, 13,14, 15, 16, 17, 18, 19, 20, ...
取新序列的第一个数5
,然后用5
把序列的5
的倍数筛掉:
7, 8, 9, 10, 11, 12, 13, 14,15, 16, 17, 18, 19, 20, ...
不断筛下去,就可以得到所有的素数。
相应的代码如下:
- ##--encoding =utf-8--
- def fun():##生成器 ,生成无限的自然数
- n = 1
- while True:
- n = n+1
- yield n
-
- def filterfun(n):##进行筛选
- return lambda x:x%n>0
-
- def primes():
- num = fun() ##初始序列 2,3,4,5,6,7,8,,,,,
- while True:
- n = next(num)
- num = filter(filterfun(n),num)
- yield n
-
- if __name__=="__main__":
- number = int(input("please input a number:"))
- j = 0 ## 每隔5个数输出
- count = 0 ##统计素数的个数
- for i in primes():
- if i<number:
- print(i)
- j=j+1
- if j%5 ==0:
- print(end = "\n")
- count = count+1
- else :
- break
- print("%d以内的素数个数:%d" %(number,count))
-
-
-
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。