模拟研究调用函数

时间:2013-04-02 09:16:59

标签: r function loops simulation

我花了好几个小时思考以下问题。我正在进行模拟研究,我想在模拟研究之外定义函数,以便能够在代码的末尾调用这些函数。

此示例说明了问题,但无法复制(下面您将找到问题的可复制示例)。我使用“metafor”包进行荟萃分析。

我想使用我在最终模拟代码之外定义的以下函数:

mat <- matrix(NA, nrow = 8, ncol = 3)

funtr.stu <- function(i) {
  for (y in 1:8) {
    mat[y,i] <- tr[[y]]$k0
  }
  return(mat)
}

“tr”是一个列表,由8次分析结果组成。我想从该列表中检索对象“k0”并将其存储到矩阵“mat”中。

在代码的下一部分(我运行模拟)中,我想调用函数并用正确的数字填充矩阵“mat”。

for (i in 1:iterations) { 
   tr.stu <- funtr.stu()
}

此代码的结果是填充矩阵,但每列中的数字相同。因此,R不是每次迭代都存储数字,而是仅存储最后一次迭代。

如何修改我的代码,使R按照我想要的方式存储输出?

一个非常简单的例子:

Mat只代表一个带数字的矩阵,res是一个我想要填充的空矩阵。

mat <- matrix(data = c(1,2,3,4,5,6), ncol = 2, nrow = 3)
res <- matrix(NA, ncol = 2, nrow = 3)

我使用函数“fun”来填充空矩阵res。

fun <- function() {
  for (i in 1:2) {
    res[y,i] <- mat[y,i]
  }
  return(res)
}

这是我想在代码的末尾放置的内容(我只想调用函数,并且使用此函数我想填充矩阵“res”)。但是,如果我使用下面的代码,则只填充第三行而不是第一行和第二行。

for (y in 1:3) {
  test <- fun()
}

提前谢谢!

1 个答案:

答案 0 :(得分:1)

这适用于您的情况。基本上,在for循环的每次迭代中返回一行。你在哪里返回整个'res'矩阵。

mat <- matrix(data = c(1,2,3,4,5,6), ncol = 2, nrow = 3)
res <- matrix(NA, ncol = 2, nrow = 3)


fun <- function() {
  for (i in 1:2) {
    res[y,i] <- mat[y,i]
  }
  return(res[y,])
}

for (y in 1:3) {
  test[y,] <- fun()
}