在函数内部使用get lapply

时间:2011-11-04 22:52:29

标签: r lapply

这可能看起来像一个过于复杂的问题,但它让我有点疯狂了一段时间。这也是为了好奇,因为我已经有办法做我需要的事情,所以并不重要。

在R中,我需要一个函数来返回一个带有所有参数和用户输入的值的命名列表对象。为此,我制作了这个代码(玩具示例):

foo <- function(a=1, b=5, h='coconut') {
    frm <- formals(foo)
    parms <- frm
    for (i in 1:length(frm))
        parms[[i]] <- get(names(frm)[i])
    return(parms)
}

所以当被问及时:

> foo(b=0)

$a
[1] 1

$b
[1] 0

$h
[1] "coconut"

这个结果很完美。问题是,当我尝试将lapply用于同一目标时,为了更高效(更优雅),它不会像我想要的那样工作:

foo <- function(a=1, b=5, h='coconut') {
    frm <- formals(foo)
    parms <- lapply(names(frm), get)
    names(parms) <- names(frm)
    return(parms)
}

问题显然在于get评估它的第一个参数(字符串,变量名称)的环境。这部分来自错误信息:

> foo(b=0)
Error in FUN(c("a", "b", "h")[[1L]], ...) : object 'a' not found

并且,因为当在.GlobalEnv环境中有正确名称的对象时,foo会返回其值:

> a <- 100
> b <- -1
> h <- 'wallnut'
> foo(b=0)
$a
[1] 100

$b
[1] -1

$h
[1] "wallnut"

显然,由于get默认情况下会在parent.frame()中进行求值,因此会搜索.GlobalEnv环境中的对象,而不是当前函数的对象。这很奇怪,因为函数的第一个版本不会发生这种情况。

我尝试了很多选项让函数get在正确的环境中进行评估,但无法正确执行(我已尝试pos=-2,0,1,2envir=NULL作为选项)。

如果有人碰巧比我更了解环境,特别是在这个“奇怪”的情况下,我很想知道如何解决这个问题。

感谢您的时间,

3 个答案:

答案 0 :(得分:9)

编辑2013-08-05

使用sapply()代替lapply(),大大简化了这一点:

foo4 <- function(a=1, b=5, h='coconut') {
    frm <- formals(sys.function())
    sapply(names(frm), get, envir=sys.frame(sys.parent(0)), simplify=FALSE)
}
foo4(b=0, h='mango')

但是,如果没有sapply() lapply()这可能是更优雅的解决方案:

foo5 <- function(a=1, b=5, h='coconut') {
    modifyList(formals(sys.function()), as.list(match.call())[-1])
}
foo5(b=0, h='mango')

原帖(2011-11-04)

铸造一下之后,这看起来是最好的解决方案。

foo <- function(a=1, b=5, h='coconut') {
    frm <- formals(foo)
    parms <- lapply(names(frm), get, envir=sys.frame(sys.parent(0)))
    names(parms) <- names(frm)
    return(parms)
}
foo(b=0, h='mango')
# $a
# [1] 1

# $b
# [1] 0

# $h
# [1] "mango"

这里有一些微妙的东西,lapply范围/评估它构造的调用的方式。详细信息隐藏在对.Internal(lapply(X, FUN))的调用中,但是为了品味,请比较这两个调用:

# With function matched by match.fun, search in sys.parent(0)
foo2 <- function(a=1, h='coconut') {
    lapply(names(formals()), 
           get, envir = sys.parent(0))
}

# With anonymous function, search in sys.parent(2)    
foo3 <- function(a=1, h='coconut') {
    lapply(names(formals()), 
           FUN = function(X) get(X, envir = sys.parent(2)))
}

foo4(a=0, h='mango')
foo5(a=0, h='mango')

答案 1 :(得分:6)

只需将当前环境转换为列表:

foo <- function(a=1, b=5, h='coconut') {
  as.list(environment())
}
foo(a = 0, h = 'mango')

答案 2 :(得分:1)

改编自@Josh O'Brien的上述解决方案,使用sapply自动为结果列表指定正确的名称(保存一行代码):

foo <- function(a=1, b=5, h='coconut') {
    frm <- formals(foo)
    parms <- sapply(names(frm), get, envir=sys.frame(sys.parent(-1)), simplify=FALSE)
    return(parms)
}