Python高级函数：map/reduce

Python内建了map()和reduce()函数。

map()

map()用法：

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

# 举个例子
def f(x):
    return x * x
r = map(f, [1,2,3,4,5,6,7,8,9])
# 由于返回的r是一个Iterator，它是惰性序列，所以可以用list()函数把整个序列都计算出来，并返回一个list
print(list(r))
# 会打印出 [1,4,9,16,25,36,49,64,81]

map()作为高阶函数，事实上它把运算规则抽象了，因此我们可以把任意一个复杂的函数作用在一个序列上，只用一行代码表示出来。

reduce()

reduce()用法：

reduce同样要传一个函数和一个序列，但是这个函数必须接收两个参数，reduce把结果继续和序列的下一个元素做累积计算。

比方说对一个序列的求和，就可以用reduce实现：

from functools import reduce
list = [1,2,3,4,5,6,7,8,9]
def sum(x,y):
    return x + y
print(reduce(sum, list))
# 45

还可以通过和map()函数的配合，把str，转成int

from functools import reduce
def char2num(s):
     digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
     return digits[s]
def fn(x, y):
     return x * 10 + y
print(reduce(fn, map(char2num, '77448')))
# 77448

通过lambda函数进一步简化：

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 char2num(s):
    return DIGITS[s]
print(reduce(lambda x, y: x * 10 + y , map(char2num, '9987986')))
# 9987986