Python如何用assignment expressions替换reduce实现列均值计算

实现方案 #

你可以完全使用赋值表达式替换functools.reduce的逻辑，不需要额外导入依赖，实现代码如下：

def get_average(xs):
    """
    该函数接收一个由n个长度为3的列表组成的列表作为输入，返回每列的均值，结果为长度为3的列表。
    """
    acc = [0, 0, 0]
    # 迭代累加每列数值，使用赋值表达式更新累加器
    [(acc := [a + dx for a, dx in zip(acc, x)]) for x in xs]
    return [v / len(xs) for v in acc]

说明 #

• 上述写法完全符合Python风格，相比原reduce + lambda的实现逻辑更直白，可读性更高
• 代码中的acc := ...就是你需要的「将赋值表达式的结果直接作为表达式使用」的写法：每次迭代过程中会先完成新累加值的计算，将值赋值给acc后，整个表达式返回新的acc值作为列表推导的元素（该临时列表仅用于触发迭代逻辑，不需要额外消费）
• 该实现不局限于输入子列表长度为3的场景，任意固定长度的子列表输入都能正常计算对应列的均值

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