【转】Python 中map、reduce、filter函数
转自:http://www.blogjava.net/vagasnail/articles/301140.html?opt=admin
介绍下Python 中 map,reduce,和filter 内置函数的方法:
一:map
map(...) map(function, sequence[, sequence, ...]) -> list
说明:
对sequence中的item依次执行function(item),执行结果输出为list。
例子:
>>> map(str, range(5)) #对range(5)各项进行str操作 ['0', '1', '2', '3', '4'] #返回列表 >>> def add(n):return n+n ... >>> map(add, range(5)) #对range(5)各项进行add操作 [0, 2, 4, 6, 8] >>> map(lambda x:x+x,range(5)) #lambda 函数,各项+本身 [0, 2, 4, 6, 8]
>>> map(lambda x:x+1,range(10)) #lambda 函数,各项+1
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> map(add,'zhoujy') ['zz', 'hh', 'oo', 'uu', 'jj', 'yy'] #想要输入多个序列,需要支持多个参数的函数,注意的是各序列的长度必须一样,否则报错: >>> def add(x,y):return x+y ... >>> map(add,'zhoujy','Python') ['zP', 'hy', 'ot', 'uh', 'jo', 'yn'] >>> def add(x,y,z):return x+y+z ... >>> map(add,'zhoujy','Python','test') #'test'的长度比其他2个小 Traceback (most recent call last): File "<stdin>", line 1, in <module> TypeError: add() takes exactly 2 arguments (3 given) >>> map(add,'zhoujy','Python','testop') ['zPt', 'hye', 'ots', 'uht', 'joo', 'ynp']
二:reduce
reduce(...) reduce(function, sequence[, initial]) -> value
说明:
对sequence中的item顺序迭代调用function,函数必须要有2个参数。要是有第3个参数,则表示初始值,可以继续调用初始值,返回一个值。
例子:
>>> def add(x,y):return x+y ... >>> reduce(add,range(10)) #1+2+3+...+9 45 >>> reduce(add,range(11)) #1+2+3+...+10 55 >>> reduce(lambda x,y:x*y,range(1,3),5) #lambda 函数,5是初始值, 1*2*5 10 >>> reduce(lambda x,y:x*y,range(1,6)) #阶乘,1*2*3*4*5 120 >>> reduce(lambda x,y:x*y,range(1,6),3) #初始值3,结果再*3 360
>>> reduce(lambda x,y:x+y,[1,2,3,4,5,6]) #1+2+3+4+5+6
21
三:filter
filter(...) filter(function or None, sequence) -> list, tuple, or string
说明:
对sequence中的item依次执行function(item),将执行结果为True(!=0)的item组成一个List/String/Tuple(取决于sequence的类型)返回,False则退出(0),进行过滤。
例子:
>>> def div(n):return n%2 ... >>> filter(div,range(5)) #返回div输出的不等于0的真值 [1, 3] >>> filter(div,range(10)) [1, 3, 5, 7, 9] >>> filter(lambda x : x%2,range(10)) #lambda 函数返回奇数,返回列表 [1, 3, 5, 7, 9] >>> filter(lambda x : not x%2,range(10)) [0, 2, 4, 6, 8] >>> def fin(n):return n!='z' #过滤'z' 函数,出现z则返回False ... >>> filter(fin,'zhoujy') #'z'被过滤 'houjy' >>> filter(lambda x : x !='z','zhoujy') #labmda返回True值 'houjy' >>> filter(lambda x : not x=='z','zhoujy') #返回:字符串 'houjy'
上面的这些例子中都用到了一个 lambda表达式,下面举个例子,更多信息见这里。
>>> a=lambda x:x+3 >>> a(2) 5 >>> a=lambda x,y:x+y >>> a(2,3) 5
四,应用map,reduce,filter
1):实现5!+4!+3!+2!+1!
#!/usr/bin/env python #-*- coding:utf-8 -*- def add_factorial(n): empty_list=[] #声明一个空列表,存各个阶乘的结果,方便这些结果相加 for i in map(lambda x:x+1,range(n)): #用传进来的变量(n)来生成一个列表,用map让列表都+1,eg:range(5) => [1,2,3,4,5] a=reduce(lambda x,y:x*y,map(lambda x:x+1,range(i))) #生成阶乘,用map去掉列表中的0 empty_list.append(a) #把阶乘结果append到空的列表中 return empty_list if __name__ == '__main__': import sys #2选1 #(一) try: n = input("Enter a Number(int) : ") result=add_factorial(n) #传入变量 print reduce(lambda x,y:x+y,result) #阶乘结果相加 except (NameError,TypeError): print "That's not a Number!" #(二) # result = add_factorial(int(sys.argv[1])) #传入变量 # print reduce(lambda x,y:x+y,result) #阶乘结果相加
结果:
zhoujy@zhoujy:~/桌面/Python/4$ python factorial.py Enter a Number(int) : 1 1 zhoujy@zhoujy:~/桌面/Python/4$ python factorial.py Enter a Number(int) : 2 3 zhoujy@zhoujy:~/桌面/Python/4$ python factorial.py Enter a Number(int) : 3 9 zhoujy@zhoujy:~/桌面/Python/4$ python factorial.py Enter a Number(int) : 4 33 zhoujy@zhoujy:~/桌面/Python/4$ python factorial.py Enter a Number(int) : 5 153 zhoujy@zhoujy:~/桌面/Python/4$
2):将100~200以内的质数挑选出来
思路:
质数是指:只有1和它本身两个因数,如2、3、5、7都是质数,即能被1和本身整除,1不是质数。
比如一个数字N,看它是否质数的话,就要看:有没有能整除【2,N】之间的数X(不包含本身),即N%X是否为0,要是没有就为质数。
所以我们要实现的算法是:拿一个数字N,去除以【2,N】之间的数X,来得到质数,即:N/2,N/3,……,N/N-2,N/N-1 ===> N/range(2,N)
#!/usr/bin/env python #-*- coding:utf-8 -*- def is_prime(start,stop): stop = stop+1 #包含列表右边的值 prime = filter(lambda x : not [x%i for i in range(2,x) if x%i == 0],range(start,stop)) #取出质数,x从range(start,stop) 取的数 print prime if __name__ == '__main__': try : start = input("Enter a start Number :") except : start = 2 #开始值默认2 try : stop = input("Enter a stop Number :") except : stop = 0 #停止数,默认0,即不返回任何值 is_prime(start,stop)
结果:
zhoujy@zhoujy:~/桌面/Python/4$ python prime.py Enter a start Number : Enter a stop Number :10 [2, 3, 5, 7] zhoujy@zhoujy:~/桌面/Python/4$ python prime.py Enter a start Number :10 Enter a stop Number :20 [11, 13, 17, 19] zhoujy@zhoujy:~/桌面/Python/4$ python prime.py Enter a start Number : Enter a stop Number : []