在并行 foreach 循环中使用 source()

Question

这是一个玩具示例来说明我的问题。

library(foreach)
library(doMC)
registerDoMC(cores=2)

foreach(i = 1:2) %dopar%{
  i + 2
}
[[1]]
[1] 3

[[2]]
[1] 4

到目前为止一切都很好......

但是，如果代码 i + 2 保存在文件 addition.R 中，并且我使用 source() 调用该文件，那么

> foreach(i = 1:2) %dopar%{
+   source("addition.R")
+ }
Error in { : task 1 failed - "object 'i' not found"

Answer 1

我无法完全复制你的玩具，但我遇到了一个类似的问题，我能够通过以下方式解决：

source(file, local = TRUE)

应该在本地环境中解析源，即识别i。

Answer 2

NiceE 的评论和 Sosel 的回答已经解决了这个问题；当调用source(file) 时，它默认为source(file, local = FALSE)，这意味着源文件中的代码正在全局环境（“用户的工作空间”）中评估，并且有，参见。 ?source。请注意，全局环境中没有变量i。解决方案是确保文件来源于调用它的环境，即使用source(file, local = TRUE)。

解决方案：

library("foreach")

y <- foreach(i = 1:2) %dopar% {
  i + 2
}
str(y)

doMC::registerDoMC(cores = 2L)
y <- foreach(i = 1:2) %dopar% {
  source("addition.R", local = TRUE)
}
str(y)

for() 循环的相同问题示例：

source() 是在全局环境中进行评估的，这与 i 所在的调用环境不同，也可以使用常规 for 循环来说明，方法是在全局环境之外的另一个环境中运行 for 循环，例如在函数内部或通过：

local({
  for(i in 1:2) {
    source("addition.R")
  }
})

给出：

Error in eval(ei, envir) : object 'i' not found

现在，上面的foreach(i = 1:2) %dopar% { source("addition.R") } 与registerDoSEQ() 一起工作的原因当且仅当 从全局环境调用，然后在调用环境中评估 foreach 迭代，即全局环境，也就是source() 使用的环境。但是，如果有人使用了local(foreach(i = 1:2) %dopar% { ... })，这也会与上面的local(for(i in 1:2) { ... }) 调用类似地失败。

总结：没有什么神奇的事情发生，但是理解它有点乏味。

Answer 3

我最终通过将 source("addition.R") 转换为函数并将变量传递给它来解决问题。我不知道为什么，但是基于source(file, local = TRUE) 的建议解决方案不起作用。