Python中如何在map内编写逻辑实现计数器自增并结合partial使用

问题根因 #

1. 你定义的计数器x在map执行过程中没有触发自增，每次lambda调用都读取的是初始值0，所以所有结果都是0+1=1
2. Python的lambda默认仅支持单表达式，你尝试的直接赋值写法不符合语法规范，即使通过海象运算符实现自增也属于带副作用的写法，不符合函数式编程范式
3. 你当前对partial的使用是正确的，没有特性理解错误，问题出在计数器的迭代逻辑上

函数式实现方案 #

函数式编程要求函数无副作用，避免依赖外部可变变量，所以不需要手动维护计数器，有两种常用的标准实现：

方案1：结合enumerate迭代下标 #

enumerate会直接返回每个元素的下标（默认从0开始），可以直接用map迭代enumerate的返回值，忽略元素本身取下标加k即可：

from functools import partial


def f(x, k):
    return x + k


def with_partial(my_list, k=1):
    func = partial(f, k=k)
    # map接收enumerate返回的(index, item)二元组，仅用index计算
    rez = list(map(lambda t: func(t[0]), enumerate(my_list)))
    print("Result with partial: " + str(rez))


if __name__ == "__main__":
    with_partial([5, 6, 7])
    with_partial(["asdf", "ghjk", "lmnop", "zyx"])

运行输出完全符合预期：

Result with partial: [1, 2, 3]
Result with partial: [1, 2, 3, 4]

方案2：用count迭代器生成序列 #

如果完全不想依赖列表下标，也可以用itertools.count生成递增序列，和输入列表同步迭代：

from functools import partial
from itertools import count


def f(x, k):
    return x + k


def with_partial(my_list, k=1):
    func = partial(f, k=k)
    # count(0)生成从0开始的无限递增序列，和输入列表长度自动匹配
    rez = list(map(lambda c, _: func(c), count(0), my_list))
    print("Result with partial: " + str(rez))

输出和方案1完全一致。

注：如果你坚持要手动维护计数器，可以通过nonlocal声明变量作用域后用海象运算符实现自增：rez = list(map(lambda it: (x := func(x)), my_list))，但这种写法引入了外部可变状态，属于典型的副作用操作，不符合函数式编程的设计原则，非常不推荐使用。

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