如何在R语言中使用for循环为数组填充不同的prob值

看起来你遇到的问题是循环里的数组初始化和索引逻辑出了问题，我来帮你理清楚并修改代码：

原代码的问题分析 #

你的代码有两个核心问题导致最后只有第一个矩阵的(1,2)位置是0.09：

1. 每次循环都重置数组：在循环内部每次都执行trans_mat <- array(0, dim = c(3, 3, 5))，这会把之前循环中修改的数组全部清空，相当于每次都从头创建一个新数组，最后自然只剩最后一次循环的赋值结果。
2. 索引逻辑错误：1:length(i)完全没必要——i是prob里的单个数值（比如0.05），length(i)返回的是1，所以你每次都是往trans_mat[1,2,1]这个位置赋值，循环到最后这个位置就被0.09覆盖了。

修改后的代码（推荐方案） #

我们需要先一次性初始化好数组，然后通过索引循环来给每个矩阵的对应位置赋值：

prob = c(0.05, 0.06, 0.07, 0.08, 0.09)
# 先创建好3x3x5的全0数组，只做一次初始化
trans_mat <- array(0, dim = c(3, 3, 5))
# 循环索引值，而不是直接循环prob的元素
for (i in 1:length(prob)) {
  # 给第i个矩阵的(1,2)位置赋值对应的prob值
  trans_mat[1, 2, i] <- prob[i]
}

验证结果 #

你可以通过以下命令检查每个矩阵的结果：

# 查看第一个矩阵
trans_mat[,,1]
# 查看第五个矩阵
trans_mat[,,5]

第一个矩阵的(1,2)位置会是0.05，第五个则是0.09，完全符合你的需求。

备选方案（适合复杂矩阵操作） #

如果每个矩阵需要做更多自定义操作，你可以先单独构建每个矩阵，再放到数组里：

prob = c(0.05, 0.06, 0.07, 0.08, 0.09)
trans_mat <- array(0, dim = c(3, 3, length(prob)))
for (i in 1:length(prob)) {
  # 单独创建一个3x3的全0矩阵
  current_mat <- matrix(0, nrow=3, ncol=3)
  # 给这个矩阵的(1,2)位置赋值
  current_mat[1,2] <- prob[i]
  # 把处理好的矩阵放到数组的第i个切片
  trans_mat[,,i] <- current_mat
}

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