Python如何实现接收lambda作为combiner参数的自定义reduce函数

原有代码的问题 #

你当前的实现思路存在几个核心错误：

• 试图通过相等判断识别combiner的运算类型，既无法覆盖所有可能的自定义运算规则，也违背了高阶函数的设计初衷
• 乘法分支的判断条件combiner == x * y存在语法错误，变量x、y未提前定义
• 乘法循环逻辑中n * i未将计算结果赋值回n，永远返回初始值1
• 硬编码加法、乘法分支的写法扩展性极差，传入其他归约规则（比如求最大值、幂运算）时完全无法工作

正确实现逻辑 #

你完全不需要使用if/else分支判断combiner的具体运算规则：Python中函数是一等公民，传入的combiner可以直接被调用，通用实现逻辑如下：

1. 优先处理单元素列表场景，直接返回列表内唯一元素即可
2. 取列表第一个元素作为累计计算的初始值
3. 从列表第二个元素开始遍历，每一步将当前累计值、当前遍历元素作为参数传入combiner，将combiner的返回值更新为新的累计值
4. 遍历完成后返回累计值，就是最终的归约结果

可运行代码 #

def reduce(combiner, lst):
    # 单元素列表直接返回
    if len(lst) == 1:
        return lst[0]
    # 初始化累计值为列表第一个元素
    accumulator = lst[0]
    # 遍历剩余元素，逐次应用combiner规则
    for item in lst[1:]:
        accumulator = combiner(accumulator, item)
    return accumulator

效果验证 #

对应你给出的测试用例，运行结果完全符合预期：

• reduce(lambda x, y: x + y, [1, 2, 3, 4]) 返回10，执行逻辑为((1+2)+3)+4
• reduce(lambda x, y: x * y, [1, 2, 3, 4]) 返回24，执行逻辑为((1*2)*3)*4
• reduce(lambda x, y: x * y, [4]) 返回4

这个实现还支持任意自定义的二元归约规则，比如传入lambda x,y: x if x>y else y就可以求列表最大值，不需要修改reduce函数本身的代码。

内容的提问来源于stack exchange，提问作者tillatastic