Python常见函数之reduce函数用法

可能突然提到reduce函数你觉得不认识，但是给你这么一个场景，假如我要你实现

def str_to_number(order):
    digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
    return digits[order]

我想让你利用上面的函数给我返回一个 12345

有人说可以啊 弄个列表每次传递一个对应的key获取value 然后append到列表里面。确实可以但是有点过于麻烦了，那么今天介绍的这个reduce函数绝效率嘎嘎的！

reduce函数

reduce把一个函数作用在一个序列[x1, x2, x3, …]上，这个函数必须接收两个参数，reduce把结果继续和序列的下一个元素做累积计算，其效果就是：

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

这么看可能不明白，下面让我来举个例子！

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

from functools import reduce

def add(a, b):
    return a + b

print(reduce(add, [1, 2, 3, 4, 5])) # 15

这就实现了将 1,2,3,4,5 利用add 函数求和

亦或者 我们可以实现 将输入的内容按照整数输出

from functools import reduce
def fn(a, b):
    return a * 10 + b

print(reduce(fn, [1, 2, 3, 4, 5]))  # 12345

这下应能看明白里面的奥妙了吧！

应用到最开始的那个问题那么我们就可以这么来做

from functools import reduce

def str_to_number(order):
    digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
    return digits[order]

def str_to_number_reduce(order):
    return reduce(lambda x, y: x * 10 + y, map(str_to_number, order))

print(str_to_number_reduce(['1','2','3','4','5'])) # 12345