创建一个函数,提供与R中的lapply相同的结果

时间:2015-04-10 22:38:40

标签: r algorithm lapply

我正在尝试分析R中lapply函数的效率,但是我没有在“base”文件夹中找到lapply的原始开发代码,所以我自己编写了一个简单的代码。

new_lapply<-function(data,f_name){
  list<-list()
  for (i in length(data)){
   list[i]<-f_name(data[,i])
  }
  return (list)
}
check<-Boston[1:10,]
lapply(check,length)

new_lapply(check,length)

Error: could not find function "f_name"

我想知道如何在函数中输入“通用函数名”,因此我可以在new_lapply函数中运行不同的函数,类似于内置的lapply()。

非常感谢你。

1 个答案:

答案 0 :(得分:1)

您应该使用match.fun()。我也做了其他一些改变。希望这能让你朝着正确的方向前进。此外,对lapply()及其效率here

有一个很好的解释
new_lapply <- function(data, f_name) {
    ## match the function in the 'f_name' argument
    f_name <- match.fun(f_name)
    ## allocate a list the same length as 'data' (ncol for data frames)
    List <- vector("list", length(data))
    for (i in seq_along(data)) {
        List[[i]] <- f_name(data[[i]])
    }
    ## if 'data' has names, carry them over to 'List' 
    if(!is.null(names(data))) 
        names(List) <- names(data)
    List
}

identical(lapply(mtcars, length), new_lapply(mtcars, length))
# [1] TRUE

um <- unname(mtcars)
identical(lapply(um, length), new_lapply(um, length))
# [1] TRUE