2013-03-21 24 views
0

我已经创建了下面的简单函数的R值多行:应用功能,使用多行

fun <- function(a,b,c,d,e){b+(c-a)*((e-b)/(d-a))} 

,我想这个功能适用于data.frame,看起来像:

> data.frame("x1"=seq(55,75,5),"x2"=round(rnorm(5,50,10),0),"x3"=seq(30,10,-5)) 
    x1 x2 x3 
1 55 51 30 
2 60 45 25 
3 65 43 20 
4 70 57 15 
5 75 58 10 

我想申请fun到每个单独的行来创建一个新的变量x4,但现在来了困难的部分(对我来说至少..):对于参数d和e我想要使用值x2和来自下一行的。因此,对于示例的第一行将意味着:fun(a=55,b=51,c=30,d=45,e=25)。我知道我可以使用mapply()来为每一行应用一个函数,但我不知道如何告诉它应该使用下一行中的某些值,或者我是否应该寻找与mapply()不同的方法?

非常感谢提前!

+0

怎么样的最后一排? 'a = 75,b = 58,c = 10',d和e是? – Arun 2013-03-21 16:05:42

+0

好点!假设'd = X2 + 2(so 60)'和'e = 0'。 (对于我的项目,最后一行实际上并不感兴趣) – Rob 2013-03-21 16:12:53

回答

6

使用mapply,但将第四列和第五列移动一行。您可以手动完成,或使用taRifx::shift

> dat 
    x1 x2 x3 
1 55 25 30 
2 60 58 25 
3 65 59 20 
4 70 68 15 
5 75 43 10 
library(taRifx) 
> shift(dat$x2) 
[1] 58 59 68 43 25 
> mapply(dat$x1, dat$x2, dat$x3, shift(dat$x2), shift(dat$x3) , FUN=fun) 
[1] 25.00000 -1272.00000 719.00000 -50.14815 26.10000 

如果你想最后一排是NA,而不是包装,使用wrap=FALSE,pad=TRUE

> shift(dat$x2,wrap=FALSE,pad=TRUE) 
[1] 58 59 68 43 NA 
+0

对不起,但是'shift(dat $ x2)'(它与OP的数据相同的x2)与你所显示的相同吗? – Arun 2013-03-21 16:23:03

+0

非常感谢@Ari这样做!为了解决最后一行的问题,我改变了代码如下:'dat $ x4 < - mapply(dat $ x1,dat $ x2,dat $ x3,c(shift(dat $ x2,wrap = F),dat $ x2 [length(dat $ x2)] + 2),c(shift(dat $ x3,wrap = F),0),FUN = fun)' – Rob 2013-03-21 16:30:37

+1

@Arun我没有意识到OP在他的随机值data.frame构造,而不指定'set.seed()',这可能是乍一看它与OP相同的原因。增加了我的'dat'的样子。 – 2013-03-21 17:59:43