在r中的编号变量中应用函数

时间:2016-11-30 02:26:07

标签: r function loops lapply

给出以下数据框('数据'):

data <- data.frame(a1 = sample(10), b1 = rep(0,10), c1 = sample(10), 
               a2 = rep(0,10), b2 = sample(10), c2 = sample(10),
               a3 = sample(10), b3 = sample(10), c3 = sample(10))

我创建的函数可以根据3个条件进行计算:

calcs <- function(a, b, c){
if ((max(a) > 0) & (max(b)>0)){
maxa <- max(a)
maxb <- max(b)
maxc <- max(c)
calc = 75*c/maxc + 12.5*a/maxa + 12.5*b/maxb
}
else if ((max(a) > 0) & (max(b)==0)){
maxa <- max(a)
maxc <- max(c)
calc = 75*c/maxc + 25*a/maxa 
}

else {
maxc <- max(c)
maxb <- max(b)
calc = 75*c/maxc + 25*b/maxb 
}
}

我想将此函数应用于数据帧,循环遍历编号变量(从1到3),而不是使用以下代码:

data$calc1 <- calcs(data$a1,data$b1,data$c1)
data$calc2 <- calcs(data$a2,data$b2,data$c2)
data$calc3 <- calcs(data$a3,data$b3,data$c3)

我在其他地方看过,显然是一个简单的问题,但是有各种各样的&#39;工具&#39;列表上的文档,lapply等我不确定最佳行动方案。

1 个答案:

答案 0 :(得分:1)

我们可以使用Map输入&#39; a&#39;&#39; b&#39;和&#39; c&#39;列的子集作为输入。

data[paste0("calc", 1:3)] <- Map(calcs, 
          data[paste0("a", 1:3)], data[paste0("b", 1:3)], data[paste0("c", 1:3)])