Question

我正在处理一些我确信必须能够使用apply系列函数编写的R代码，但我无法弄清楚如何。我有一个包含多列的数据框，我想调用一个函数，函数的输入使用数据帧中的多个列。假设我有这个数据和函数f：

data<- data.frame(T=c(1,2,3,4), S=c(3,7,8,4), K=c(5,6,11,9))
data
V<-c(0.1,0.2,0.3,0.4,0.5,0.6)

f<-function(para_h,S,T,a,t,b){
  r<- V
  steps<-T
  # Recursive form: Terminal condition for the A and B at time T
  A_T=0
  B_T=0
  A=c()
  B=c()
  #  A and B a time T-1
  A[1]= r[steps]*a 
  B[1]= a*para_h[5]+ ((para_h[4])^(-2))         
  # Recursion back to time t      
  for (i in 2:steps){
    A[i]= A[i-1]+ r[steps-i+1]*a + para_h[1]*B[i-1]
    B[i]= para_h[2]*B[i-1]+a*para_h[5]+ (para_h[4]^(-2))
}
  f = exp(log(S)*a + A[t] + B[t]*b )

  return(f)
}

此功能适用于某些特定值：

> para_h<-c(0.1,0.2,0.3,0.4,0.5,0.7)
> f(para_h,S=3,T=2,a=0.4,t=1,b=0.1)
[1] 3.204144

我想将函数应用于数据框中的每个列S和T.所以，我的代码看起来像：

mapply(function(para_h,S,T,a,t,b) f(para_h,S,T,a,t,b) ,para_h,S=data$S,T=data$T,a=0.4,t=1,b=0.1)

这会出错：

> mapply(function(para_h,S,T,a,t,b) f(para_h,S,T,a,t,b) ,para_h,S=data$S,T=data$T,a=0.4,t=1,b=0.1)
Error in A[i] = A[i - 1] + r[steps - i + 1] * a + para_h[1] * B[i - 1] : 
  replacement has length zero

我很确定问题是：“步骤”是矢量。真的很感激优雅的解决方案。我希望这有一定道理，任何建议都会受到高度赞赏。

Answer 1

一些事情：

1）您的函数的每次调用都需要完整的para_h向量，但在mapply代码中，它一次只会收到一个值，所以您可能想要这样的内容：

 mapply(function(S,T) f(para_h,S,T,a=0.4,t=1,b=0.1), data$S, data$T)

或者这个：

 apply(data,1,function(d) f(para_h,d['S'],d['T'],a=0.4,t=1,b=0.1))

2）当T==1（data的第一行中的情况）时，您的函数会抛出错误，因此您可能需要修改示例数据集才能运行此代码。

R mapply（）关于具有递归形式的特定函数（使用for）

1 个答案: