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

Question

我想在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()。

非常感谢。

Answer 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