时间戳

首页 > 解决方案 > 如何更快地运行for循环R程序？

r - 如何更快地运行for循环R程序？

问题描述

我正在使用以下内容r code来计算每个的左侧和右侧的对数似然i = 1,2,...,200。

但是我想对大量生成的数据集执行此过程，例如a = 10000并多次迭代整个循环1000。如何加快以下程序？我可以使用apply函数而不是for函数吗？

先感谢您！

n1 = 100
n2 = 100
a = 1000 
n= n1 + n2
# number of simulated copies of y
  sim.data = matrix(NA, nrow = n, ncol = a)
  for (i in 1:a) {
    #for(j in 1:a){
    sim.data[,i] = c(rnorm(n1, 2, 1), rnorm(n-n1, 4, 1))
    #}
  }
  dim(sim.data)


  # Compute the log-likelihood
  B = ncol(sim.data)
  loglike_profb = matrix(NA, n - 1, B)
  for (j in 1:B) {
    for (i in 1:(n - 1)) {
      loglike_profb[i, j] = -0.5*(sum(((sim.data[1:i,j]) - mean(sim.data[1:i,j]))^2) + sum(((sim.data[(i + 1):n,j]) - mean(sim.data[(i +1):n,j]))^2))
    }
  }

标签： rfor-loopapplysapply

解决方案

您可以将 loglike_profb 的计算放入一个函数中，然后使用mapply

loglike_profb_func <- function(i,j){
  -0.5*(sum(((sim.data[1:i,j]) - mean(sim.data[1:i,j]))^2) + sum(((sim.data[(i + 1):n,j]) - mean(sim.data[(i +1):n,j]))^2))
}
mapply(loglike_profb_func, rep(1:(n-1),B), rep(1:B,(n-1)))

推荐阅读