最近刷leetcode,发现使用map()/reduce()这两个函数往往可以四两拨千金.但是使用起来却不简单,记录下来,免得忘记.
一 map(): 形式是map(function, sequence)
接受两个参数,第一个是函数,第二个是个可迭代对象.返回也是一个可迭代对象
第二个对象的元素依次带入第一个函数里面进行计算:
def func(x):
return x*x
m=map(func,range(1,4))
print(list(m))
#输出:[1, 4, 9, 16]
如果传入多个对象,则按元素最少的为准:
def func(x,y):
return x+y
m=map(func,range(1,8),range(3,6))
print(list(m))
#输出:[4, 6, 8]
二 reduce(): 要先导入库 functors .
形式是reduce(function, sequence,[initial])
接受三个参数,第一个是函数,第二个是个可迭代对象,第三个是初始值,返回是一个值
第一个函数必须有两个参数,必需.
第二个序列是个可迭代对象,如tuple ,list ,dictionary, string等,必需.
第三个初始值计算时作为函数第一个参数,如果序列为空,则返回初始值,非必需,如果没有初始值,则函数第一个参数是序列第一个元素,第二个参数是序列第二个元素.(有妙用!)
例如,
from functools import reduce
res = reduce(lambda x, y: x+y, [1, 2, 3, 4, 5])
print(res)
#1+2+3+4+5
#输出:15
from functools import reduce
res = reduce(lambda x, y: x+y, [1, 2, 3, 4, 5],1)
print(res)
#1+1+2+3+4+5
#输出:16