如何将字符串作为参数传递给期望R中的变量的函数

时间:2016-09-11 13:23:55

标签: r data.table nse

第一次调用函数f有效,第二次调用没有。如何将字符串(" v")传递给函数f,以便函数按预期工作?

library(data.table)

f<-function(t,x) t[,deparse(substitute(x)),with=F]

dat<-data.table(v="a")

f(dat,v)
#    v
# 1: a

f(dat,eval(parse(text="v")))
# Error in `[.data.table`(t, , deparse(substitute(x)), with = F) : 
#   column(s) not found: eval(parse(text = "v")) 

1 个答案:

答案 0 :(得分:3)

它不再是一个单行,但你可以测试你传入的内容:

library(data.table)
library(purrr)

dat <- data.table(v="a")

f <- function(dt, x) {

  # first, see if 'x' is a variable holding a string with a column name

  seval <- safely(eval)
  res <- seval(x, dt, parent.frame())

  # if it is, then get the value, otherwise substitute() it

  if ((!is.null(res$result)) && inherits(res$result, "character")) {
    y <- res$result
  } else {
    y <- substitute(x)
  }

  # if it's a bare name, then we deparse it, otherwise we turn
  # the string into name and then deparse it

  if (inherits(y, "name")) {
    y <- deparse(y) 
  } else if (inherits(y, "character")) {
    y <- deparse(as.name(x))
  }

  dt[, y, with=FALSE]

}

f(dat,v)
##    v
## 1: a

f(dat, "v")
##    v
## 1: a

V <- "v"
f(dat, V)
##    v
## 1: a

f(dat, VVV)
#> throws an Error

我将其从t切换为dt因为我不喜欢使用内置函数的名称(如t())作为变量名称,除非我真的需要。它可能会在较大的代码块中引入微妙的错误,这可能会让调试变得令人沮丧。

我还会在safely()函数之外移动f()调用,以便在每次运行f()时保存函数调用。如果您愿意,可以使用旧式try(),但是您必须检查可能会在某一天中断的try-error。你也可以tryCatch()包装它,但safely()方式对我来说似乎更清晰。