第一次调用函数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"))
答案 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()
方式对我来说似乎更清晰。