在R中用foreach替换for循环时的维度匹配问题

原始循环逻辑 #

以下是需要并行化的R代码逻辑，实际场景中c(i,j)为矩阵乘法%*%运算，此处仅作维度演示：

res <- array(0, c(4, 3, 2))

for (i in 1:4) {
    for (j in 1:3) {
      res[i,j,] <- c(i,j)
    }
}

预期输出 #

该代码生成的三维数组如下：

, , 1

     [,1] [,2] [,3]
[1,]    1    1    1
[2,]    2    2    2
[3,]    3    3    3
[4,]    4    4    4

, , 2

     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    1    2    3
[3,]    1    2    3
[4,]    1    2    3 

尝试的并行代码及错误输出 #

使用两层foreach尝试并行，但输出维度不符合预期：

res2 <- foreach(i=1:4, .combine='rbind') %dopar% {
  foreach(j=1:3 , .combine='c')%do% {
    c(i,j)
  }
} 

错误输出（维度扁平化，缺失三维结构）：

result.1    1    1    1    2    1    3
result.2    2    1    2    2    2    3
result.3    3    1    3    2    3    3
result.4    4    1    4    2    4    3

需求：不新增迭代器（第三维度由每步返回的向量长度确定），实现与原始循环一致的三维数组输出。

解决方案 #

调整合并逻辑与维度重构 #

问题核心在于原并行代码的合并方式将所有向量扁平化，破坏了三维结构。修改思路是让内层循环返回对应i行的二维矩阵，外层合并后再重构为三维数组：

library(foreach)
library(doParallel)

# 初始化并行集群
cl <- makeCluster(detectCores())
registerDoParallel(cl)

res2 <- foreach(i=1:4, .combine='rbind') %dopar% {
  # 内层循环合并为2×3矩阵，转置后得到3×2矩阵（对应res[i,,]的结构）
  inner_mat <- foreach(j=1:3, .combine='cbind') %do% {
    c(i,j)
  }
  t(inner_mat)
}

# 将合并后的二维矩阵重构为三维数组
res2 <- array(res2, dim = c(4, 3, 2))

# 关闭并行集群
stopCluster(cl)

代码说明 #

1. 内层循环用.combine='cbind'：每个c(i,j)是长度为2的向量，合并后得到2×3矩阵，转置后变为3×2矩阵，正好匹配原始循环中res[i,,]的结构。
2. 外层循环用.combine='rbind'：将每个i对应的3×2矩阵按行合并，得到4×6的二维矩阵。
3. 最后通过array()指定维度c(4,3,2)，直接重构为目标三维数组。

验证结果 #

运行后res2的输出与原始res完全一致，符合预期。

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