R语言如何通过for循环将矩阵值赋值给三维array

问题原因 #

1. 循环参数设置错误：你写的for (i in 4)仅会遍历i=4这一个值，所以只有第三维度的第4层被赋值，1-3层全部保留了初始化的0值
2. 未单独处理特殊层：你预期的第三维度第3层的矩阵结构和其他层不同，需要单独赋值逻辑

修复方案 #

方法1：调整循环逻辑（可读性更高）

m <- array(0, c(2,2,4))
# 提前定义两类矩阵
mat_normal <- matrix(1:4, nrow=2, ncol=2)
mat_special <- matrix(c(1,2,4,3), nrow=2, ncol=2)

# 遍历全部4个第三维度切片
for (i in 1:4) {
  if (i == 3) {
    m[,,i] <- mat_special
  } else {
    m[,,i] <- mat_normal
  }
}

方法2：向量化直接生成（更简洁，无需循环）

利用R数组按列优先填充的特性，直接拼接所有元素生成目标数组：

m <- array(
  data = c(rep(1:4, 2), c(1,2,4,3), 1:4),
  dim = c(2,2,4)
)

两种方案运行后打印m都能得到你预期的结果。

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