2017-07-03 169 views
1

我正在努力编写这个递归程序,并想知道是否有人可以提供帮助。递归循环R

我想这个代码回归方程:

for k=1,2,... 

beta(k)=k-sum_(i=0)^(k-1)Kchoosei*beta(i)*exp(-i(k-i)) 

我已经做到了在R上的手工方式,但希望把它放在一个函数。

beta0<-0 

beta1<-1-choose(1,0)*beta0*exp(-0*lambdaL*(1-0)) 
beta2<-2-choose(2,0)*beta0*exp(-0*lambdaL*(2-0))-choose(2,1)*beta1*exp(-1*lambdaL*(2-1)) 
beta3<-3-choose(3,0)*beta0*exp(-0*lambdaL*(3-0))-choose(3,1)*beta1*exp(-1*lambdaL*(3-1))-choose(3,2)*beta2*exp(-2*lambdaL*(3-2)) 
beta4<-4-choose(4,0)*beta0*exp(-0*lambdaL*(4-0))-choose(4,1)*beta1*exp(-1*lambdaL*(4-1))-choose(4,2)*beta2*exp(-2*lambdaL*(4-2))-choose(4,3)*beta3*exp(-3*lambdaL*(4-3)) 
+0

lambdaL没有定义,而不是在你的一般方程 –

回答

1

您可以为总和定义第二个循环。请注意,此处的索引从1开始,而不是0,这会导致“索引偏移”。

beta = numeric() 
beta[1] <- 0 

for (k in 1:10){ 
    beta[k+1] <- k 
    for (i in 0:(k-1)) 
    beta[k+1] <- beta[k+1] - choose(k, i)*beta[i+1]*exp(-i*(k-i)) 
} 

beta 
# [1] 0.000000 1.000000 1.264241 2.080705 3.247551 4.528104 5.748673 
# [8] 6.876234 7.941197 8.972749 9.987645 
0

我觉得你需要2个函数,因为你需要测试版的每个以前的值作为输入,而你只需要一个输出。这是我提出(与您lambaL调整,因为在你的帖子有些含糊,它与LambdaL == 1作品):

beta_vec <- function(k){ 
    if(k == 0) 0 else { 
    beta_vec_old <- beta_vec(k-1) 
    c(beta_vec_old,sum(sapply(0:(k-1),function(i){1-choose(k,i)*beta_vec_old[i+1]*exp(-i*(k-i))}))) 
}} 

beta <- function(k){ 
    tail(beta_vec(k),1) 
} 

# > beta_vec(5) 
# [1] 0.000000 1.000000 1.264241 2.080705 3.247551 4.528104 
# > beta(5) 
# [1] 4.528104 

(编辑在代码错字)