apply不会导致相同的向量,例如替换为for循环

时间:2017-03-22 13:47:52

标签: r for-loop apply lapply sapply

功能" trio"举例说明具有大量计算的函数,它从数据框(df)中获取某些列,并为计算获取三个值:

# example: data
df <- data.frame(C1= c(1,2,3,4,5,6),C2= c(5,4,3,2,1,6),C3= c(9,1,5,2,7,4))

# example: function
trio  <- function(a,b,c){
             df2 <- df %>%
             mutate(., 
                 df$C1 * a,
                 df$C2 * b,
                 df$C2 * c 
              ) 
df2$calc <- df2[3] * df2[4]/df2[5]
max(df2$calc)
}

找到哪个值组合产生最佳输出(= max(df2 $ calc),创建具有值组合的矩阵:

pp <- expand.grid(parameter_A = round(seq(min, max, length.out = count),3),
                  parameter_B = round(seq(min, max, length.out = count),3), 
                  parameter_C = round(seq(min, max, length.out = count),3))

替换for循环

max.all<- c()
for(w in 1:nrow(pp)){
    max.all[w] <-  findWinZ(pp[w,1],pp[w,2],pp[w,3])  
    }

我的目的是使用Apply-Family的一个功能(sapply,lapply),例如

Version1: res2 <- sapply(pp, function(a,b,c)  findWinZ(length(pp[,1]),length(pp[,2]),length(pp[,3])))  

Version: res3 <- sapply(pp, function(a,b,c)  length(findWinZ(pp[w,1],pp[w,2],pp[w,3])))  

唉,apply()循环的结果是错误的。我需要的是具有所有结果的向量,即与for循环相同的输出。 请问,任何人都可以更正我的apply()代码吗?非常感谢!

1 个答案:

答案 0 :(得分:1)

如果问题是如何将for循环重写为sapply,那么:

res4 <- sapply(1:nrow(pp), function(w) findWinZ(pp[w,1],pp[w,2],pp[w,3]))