将未引用的函数参数传递给data.table中的i

时间:2019-11-08 18:29:48

标签: r function data.table parameter-passing

我正在为我的同事创建一个函数,以轻松分析一些预先指定的数据集。我希望他们能够传递“ i”参数来选择行。为了使它更可靠,我希望能够传递带或不带引号的参数。

这里是一个使用引号的示例……

x <- data.table(x = c(1,1,1,2,2,2,3,3,3), 
  sex = c("M", "M", "F","M", "F","M", "F","M", "F"))
test.function <- function(my.dt, ...){
  where <- parse(text = paste0(list(...))) 
  my.dt <- my.dt[eval(where), ]
  return(my.dt)
}
tmp <- test.function(x, 'x==3 | sex=="F"')
head(tmp) 

如果我删除引号,它会说“找不到对象'sex'” ...

x <- data.table(x = c(1,1,1,2,2,2,3,3,3), 
  sex = c("M", "M", "F","M", "F","M", "F","M", "F"))
test.function <- function(my.dt, ...){
  where <- parse(text = paste0(list(...))) 
  my.dt <- my.dt[eval(where), ]
  return(my.dt)
}
tmp <- test.function(x, x==3 | sex=="F")
head(tmp)

除上述内容外,我还多次尝试使用引号,替换和rlang:类似此失败的示例……

x <- data.table(x = c(1,1,1,2,2,2,3,3,3), 
                sex = c("M", "M", "F","M", "F","M", "F","M", "F"))
test.function <- function(my.dt, ...){
  where <- rlang::quos(...)
  my.dt <- my.dt[!!where, ]
  return(my.dt)
}
tmp <- test.function(x, 'x==3 | sex=="F"')
head(tmp, 10)

除了解决方案的任何想法外,如果有人可以将我引向simpleton的在线资源以理解符号/表达式/引号/等,我将不胜感激。当我研究这个问题时,我意识到我在理解R的工作原理方面存在严重差距。

顺便说一句,我已经阅读了以下内容,但都没有帮助我:Passing function argument to data.table i    和    Passing multiple arguments to data.table inside a function    

1 个答案:

答案 0 :(得分:1)

可能还有其他(更好)的选项,但是您可以将其包装在tryCatch中,并使用bquote作为未引用的参数

test.function <- function(my.dt, ...){
    where <- tryCatch(parse(text = paste0(list(...))),  error = function (e) parse(text = paste0(list(bquote(...)))))
    my.dt <- my.dt[eval(where), ]
    return(my.dt)
  }

tmp <- test.function(x, 'x==3 | sex=="F"')
head(tmp) 
   x sex
1: 1   F
2: 2   F
3: 3   F
4: 3   M
5: 3   F

tmp <- test.function(x, x==3 | sex=='F')
head(tmp)
   x sex
1: 1   F
2: 2   F
3: 3   F
4: 3   M
5: 3   F