使用apply R的递归循环

时间:2018-02-08 10:46:46

标签: r for-loop recursion apply

我有两个循环,我想使用一些应用函数进行转换,以减少计算时间。第一个似乎很容易做到,问题出现在第二个,因为S在每次迭代中都有自己的值更新。

S0   = 100
a    = 0.00016
b    = 0.0126
sim    = 10000 
drifts = 1000
Si = rep(0,sim)

for(i in (1:sim))
{
 S =  S0
 for (j in (1:drifts))
  {
   z = rnorm(1, mean = 0, sd = 1)
   S = S * exp(a + b*z)
  }
  Si[i] =S 
}

有人可以帮忙吗?

2 个答案:

答案 0 :(得分:1)

calc_s <- function(S, i = 1) {
  S <- S * exp(a + b * rnorm(1, mean = 0, sd = 1))
  return (if (i < drifts) calc_s(S, i + 1) else S)}
S2 <- sapply(1:sim, function(x) {
  calc_s(S0)
})

虽然

并不快

答案 1 :(得分:1)

如下所示的代码如何?我用产品替换了j-loop,我认为数学是正确的。

sapply(1:sim,function(x) S0*prod(exp(a + b*rnorm(drifts))))

它也明显更快:

> system.time(for(i in (1:sim))
+ {
+   S =  S0
+   for (j in (1:drifts))
+   {
+     z = rnorm(1, mean = 0, sd = 1)
+     S = S * exp(a + b*z)
+   }
+   Si[i] =S 
+ }
+ )
   user  system elapsed 
  23.29    0.02   23.34 
> 
> system.time(Si<-sapply(1:sim,function(x) S0*prod(exp(a + b*rnorm(drifts)))))
   user  system elapsed 
   1.76    0.00    1.76