Python中的函数式编程-高阶函数map/reduce

时间:2022-10-13 00:09:31

map

map()函数接收两个参数,一个是函数,一个是序列,map将传入的函数依次作用到序列的每个元素,并把结果作为新的list返回。

比如我们有一个函数f(x)=x^2,要把这个函数作用在一个list[1,2,3,4,5,6,7,8,9]上,就可以用map()实现如下:

>>>def f(x):
...  return x*x
...
>>>map(f,[1,2,3,4,5,6,7,8,9])
[1,4,9,16,25,36,49,64,81]

map()传入的第一个参数是f,即函数对象本身。

map()作为高阶函数,事实上它把运算规则抽象,因此,我们不但可以计算简单的f(x)=x^2,还可以计算任意复杂的函数,比如把这个list转为字符串:


>>>map(str,[1,2,3,4,5,6,7,8,9])
['1','2','3','4','5','6','7','8','9']

reduce

reduce把一个函数作用于一个序列[x1,x2,x3....]上,这个函数必须接收两个参数,reduce把结果和序列的下一个元素做累计计算,其效果就是:


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

比如对一个序列求和,就可以用reduce实现:


>>>def add(x,y):
...  return x+y
...
>>>reduce(add,[1,3,5,7,9])
25

当然求和运算可以直接用Python内建函数sum(),没必要动用reduce。

但是如果把序列[1,3,5,7,9]变换成整数13579,reduce就派上用场了:


>>>def fn(x,y):
...  return x*10+y
...
>>>reduce(fn,[1,3,5,7,9])
13579

这个例子本身没有多大用处,但是,如果考虑到字符串str也是一个序列,对上面例子稍加改动,配合map(),我们就可以写出把str转换为int的函数:


>>>def fn(x,y):
...  return x*10+y
...
>>>def char2num(S):
...  return {'0':0,'1':1,'2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}[s]
...
>>>reduce(fn,map(char2num,'13579'))
13579

整理成一个str2int的函数就是:


def str2int(s):
 def fn(x,y):
  return x*10+y
 def char2num(s):
  return {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}[s]
 retrun reduce(fn,map(char2num,s))