R-如何将设置操作表达式编程为函数的参数



如果我有函数 f


f <- function(a = character(0),
              b = character(0),
              c = character(0),
              condition = "a + b + c") {
  if(condition == "a + b + c")
    return(union(union(a, b), c))
  if(condition == "(a * b) + c")
    return(union(intersect(a, b), c))
  if(condition == "(a - b) + c")
    return(union(setdiff(a, b), c))
  if(condition == "(a - b) - c")
    return(setdiff(setdiff(a, b), c))
  # etc...
}

f(a = c('1', '2', '3'),
  b = c('2', '3', '4'),
  c = c('3', '4'),
  condition = "a + b + c")
#> [1] "1" "2" "3" "4"
f(a = c('1', '2'),
  b = c('2', '3'),
  c = c('2', '3', '4'),
  condition = "(a * b) + c")
#> [1] "2" "3" "4"
f(a = c('1', '2'),
  b = c('2', '3'),
  c = c('2', '3', '4'),
  condition = "(a - b) - c")
#> [1] "1"

如何处理编码该参数的问题condition,该参数应该在输入向量上定义设定的操作表达式?

在我的示例中,我使用了简单的字符串来体现预期的行为,但是该解决方案不能很好地扩展。我需要使用语言对象(?!)...

我需要一种简单的方法:

  1. 指定此表达式(字符串可能是一个好对象?);
  2. 以某种方式解析并确保它转化为有效的集合操作表达式;
  3. 通过应用上述集合操作表达式来计算结果。

您可以做的一件事就是解析表达式,然后用适当的功能替换+-*。然后,您可以避免这种表达。例如

f <- function(a = character(0),
              b = character(0),
              c = character(0),
              condition = "a + b + c") {
  parsed_cond <- parse(text=condition)[[1]]
  translated_expr <- do.call("substitute", list(
      parsed_cond,
      list(`+`=quote(union),
           `-`=quote(setdiff),
           `*`=quote(intersect))
   ))
   eval(translated_expr)
}

这返回您在示例中给出的值

最新更新