python高阶函数——map/reduce

python 内置了map()和reduce()函数

1、map()函数

map()函数接收两个参数，一个是函数，一个是可迭代对象Iterable，map将传入的函数依次作用于序列的每一个元素。并把结果作为一个迭代器Iterator返回。

比如，函数f(x) = x²，要把这个函数作用在一个list[1,2,3,4,5,6]上：

>>> def f(x):
...     return x*x
...
>>> map(f,[1,2,3,4,5,6])
<map object at 0x00E0BCF0>
>>> m = map(f,[1,2,3,4,5,6])
>>> list(m)
[1, 4, 9, 16, 25, 36]

此例中，map传入的第一个参数是f函数，由于结果m是一个Iterator，可以通过list()函数把整个序列都计算出来并显示一个list。

再比如：把list中所有的数字转化为字符串：

>>> list(map(str,[1,2,3,4,5,6]))
['1', '2', '3', '4', '5', '6']

2、reduce()函数

reduce()函数内置在模块functools中。reduce也是把一个函数作用在一个序列上，只是这个传入的函数必要要接收2个参数，reduce再把这个传入的函数运算的结果和序列的下一个元素一起传入这个函数中，效果相当于：

reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)

例子1：对一个list求和：

>>> from functools import reduce
>>> def add(x,y):
...     return x+y
...
>>> reduce(add,[1,2,3,4,5,6])
21

例子2：把list[1,2,3,4,5,6]转换成整数123456：

>>> from functools import reduce
>>> def listToInt(x,y):
...     return x*10 + y
...
>>> reduce(listToInt,[1,2,3,4,5,6])
123456

例子3：把一个str字符串类型的对象转换成int整数类型的对象：

>>> from functools import reduce
>>> DIGITS={'0':0,'1':1,'2':2,'3':3,'4':4,'5':5,'6':6,'7':7,'8':8,'9':9}
>>> def str2int(s):
...     def fx(x,y):
...             return x*10 +y
...     def char2num(s):
...             return DIGITS[s]
...     return reduce(fx,map(char2num,s))
...
>>> str2int('1234567')
1234567

例子4：利用mao()函数，把用户输入的不规范的英文名字，变为首字母大写，其他的小写。输入['abc','xYz']，输出['Abc','Xyz']：

>>> def normal(n):
...     return n[0].upper()+n[1:].lower()
...
>>> l=['abc','xYz']
>>> list(map(normal,l))
['Abc', 'Xyz']

这里没有考虑开端为空字符的情况。

例子5：利用reduce()函数，编写一个函数，将接收到的list各个元素求积：

 

>>> from functools import reduce
>>> def pro(x,y):
...     return x*y
...
>>> reduce(pro,[2,4,6,8])

例子6：使用reduce和map函数，编写一个str2float函数，把字符串‘12.34’转换为浮点型12.34：

>>> from functools import reduce
>>> def str2float(s):
...     def pro(x,y):
...             return x*10 +y
...     def char2num(s):
...             return DIG[s]
...     DIG = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
...     num = s.split('.')
...     intnum = reduce(pro,map(char2num,num[0]))
...     floatnum = reduce(pro,map(char2num,num[1]))/(pow(10,len(num[1])))
...     return intnum + floatnum
...
>>> str2float('12.34')
12.34