R语言循环调用闭包累加器的异常原因及解决方法

问题现象 #

存在如下案例：

acc = \(a) \(b) base::list(a = a, b = b)
for (x in c("A","B")) {acc = acc(x)}
acc

运行结果为：

$a
[1] "B"

$b
[1] "B"

但预期应与以下分步调用结果一致：

acc = \(a) \(b) base::list(a = a, b = b)
# for (x in c("A","B")) {acc = acc(x)}
acc = acc("A")
acc = acc("B")
acc

即预期结果为：

$a
[1] "A"

$b
[1] "B"

更多案例 #

acc = \(x) \(y) \(z) base::list(x = x, y = y, z = z)
for (i in c("A","B","C")) {acc = acc(i)}
acc

运行结果为：

$x
[1] "C"

$y
[1] "C"

$z
[1] "C"

但预期应与以下分步调用结果一致：

acc = \(x) \(y) \(z) base::list(x = x, y = y, z = z)
# for (i in c("A","B","C")) {acc = acc(i)}
acc = acc("A")
acc = acc("B")
acc = acc("C")
acc

即预期结果为：

$x
[1] "A"

$y
[1] "B"

$z
[1] "C"

while循环存在相同问题 #

acc = \(a) \(b) base::list(a = a, b = b)

.iter = c("A","B")
.i = 0
while (.i < length(.iter)) {acc <<- acc <- acc(.iter[.i + 1]); .i = .i + 1}

print(acc)

问题成因 #

这个问题的核心是R语言的延迟求值机制与闭包的环境引用捕获共同作用导致的：

1. 闭包的环境绑定：嵌套匿名函数（如\(a) \(b) list(a=a, b=b)）的内层函数会捕获外层函数的执行环境。循环中每次调用acc(x)时，返回的闭包并未立即锁定当前x的数值，而是保留了对循环变量x所在环境的引用。
2. 延迟求值的触发时机：R默认采用延迟求值，函数参数不会在调用时立即计算，而是在实际需要使用该值时才去环境中查找。循环结束后，循环变量的最终值（如第一个案例的"B"、第二个案例的"C"）会被所有闭包共享引用，因此最终生成的列表所有元素都取到循环变量的最终值。
3. 分步调用的差异：分步调用时，每次传入的是字面量参数（如acc("A")），闭包捕获的是已确定的常量值，而非变量引用，因此能得到预期结果。

解决方法 #

受base::Reduce和purrr:::reduce_impl启发，可以使用base::forceAndCall强制提前计算参数值，避免闭包捕获变量引用：

acc = \(a) \(b) base::list(a = a, b = b)
for (x in c("A","B")) {acc <- base::forceAndCall(2, \(f, ...) f(...), acc, x)}
acc

base::forceAndCall的官方说明：

若函数是闭包，调用时先强制计算指定数量的前置参数再执行调用。
forceAndCall旨在帮助定义高阶函数（如apply），当被应用函数返回捕获参数的闭包时，让其行为更合理。

内容的提问来源于stack exchange，提问作者ypa y yhm