Python内建map()和reduce()函数
map()函数接收两个参数,一个是函数,一个是序列,map将传入的函数依次作用到序列的每个元素,并把结果作为新的list返回。
map()传入的第一个参数是f,即函数对象本身。
比如有一个函数f(x)=x2,要把这个函数作用在一个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]
把list所有数字转为字符串:
map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9])
结果:
['1', '2', '3', '4', '5', '6', '7', '8', '9']
reduce()函数也必须接收两个参数,把一个函数作用在一个序列[x1, x2, x3…]上,把结果继续和序列的下一个元素做累积计算,其效果就是:
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)
把序列[1, 3, 5, 7, 9]变换成整数13579
def fn(x, y):
return x * 10 + y
reduce(fn, [1, 3, 5, 7, 9])
结果:
13579
字符串str也是一个序列,reduce()配合map(),把str转换为int的函数:
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]
return reduce(fn, map(char2num, s))
用lambda函数进一步简化成:
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]
def str2int(s):
return reduce(lambda x,y: x*10+y, map(char2num, s))
利用map()函数,把用户输入的不规范的英文名字,变为首字母大写,其他小写的规范名字。输入:[‘adam’, ‘LISA’, ‘barT’],输出:[‘Adam’, ‘Lisa’, ‘Bart’]。
def upperFirstOnly(s):
return s[0].upper()+s[1:].lower()
def op(s):
return map(upperFirstOnly,s)
op(['adam', 'LISA', 'barT'])
结果:
['Adam', 'Lisa', 'Bart']
接收一个list并利用reduce()求积:
def prop(s):
return reduce(lambda x,y:x*y,s)
prod([1,2,3,4,5])
结果:
120