如何使用非标准计算来分析和修改 R 表达式



我想将 R 表达式解析为列表,并选择性地修改它的各个方面,然后再最终将其转换为 json 对象。 例如,我试图得到这样的东西:

{"op": "=",
      "content": {
          "lhs": "gender",
          "rhs": ["male"]
      }
}

我将从一个 R 表达式开始,如下所示:

gender == "male"

我可以使用pryr::ast来获取树的文本版本,但我想将其作为列表获取,例如:

op: "=="
  [[1]]: "gender"
  [[2]]: "male"

列表"格式"的细节并不那么重要,只是要清楚。 我只是想得到一个可计算和可修改的 R 表达式解析树。

像这样的东西是你要找的吗?

expr <- quote(gender == "male")
expr[[1]]
# `==`
expr[[2]]
# gender
expr[[3]]
# "male"
expr[[3]] <- "female"
expr
# gender == "female"

这是请求输出部分的一种方法,使用对我评论中引用的方法的修改。这是基于Hadley的pkg:pryr。有关中缀运算符的列表,请参阅 ?Ops。我已经看到了lhsrhs定义的函数......哈德利高级编程文本中的IIRC。显然,标记为"ops"的唯一函数将是中缀数学和逻辑,但是可以使用?groupGeneric页面中的其他列表对Math(),Complex()和Summary()函数进行更完整的标记:

call_tree2(quote(gender == "male")) # relabeling of items in pryr-functions
#--------
 - call:
   - `op: ==
   - `gender
   -  "male" 

函数定义如下:

library(pryr) # also loads the stringr namespace
# although the `tree` function is not exported, you can see it with:
pryr:::tree   # now for some hacking and adding of logic
tree2<-
function (x, level = 1, width = getOption("width"), branch = " - ") 
{
    indent <- str_c(str_dup("  ", level - 1), branch)
    if (is.atomic(x) && length(x) == 1) {
        label <- paste0(" ", deparse(x)[1])
        children <- NULL
    }
    else if (is.name(x)) {
        x <- as.character(x)
        if (x == "") {
            label <- "`MISSING"
        }
        if (x %in% c("+", "-", "*", "/", "^", "%%", "%/%",
"&", "|", "!","==", "!=", "<", "<=", ">=", ">") ) {
             label <- paste0("`op: ", as.character(x))}
        else {
            label <- paste0("`", as.character(x))
        }
        children <- NULL
    }
    else if (is.call(x)) {
        label <- "call:"
        children <- vapply(as.list(x), tree2, character(1), level = level + 
            1, width = width - 3)
    }
    else if (is.pairlist(x)) {
        label <- "[]"
        branches <- paste("", format(names(x)), "=")
        children <- character(length(x))
        for (i in seq_along(x)) {
            children[i] <- tree2(x[[i]], level = level + 1, width = width - 
                3, branch = branches[i])
        }
    }
    else {
        if (inherits(x, "srcref")) {
            label <- "<srcref>"
        }
        else {
            label <- paste0("", typeof(x), "")
        }
        children <- NULL
    }
    label <- str_trunc(label, width - 3)
    if (is.null(children)) {
        paste0(indent, label)
    }
    else {
        paste0(indent, label, "n", paste0(children, collapse = "n"))
    }
}
environment(tree2)<-environment(pryr:::tree)

现在用call_tree2来调用它:

pryr::call_tree
call_tree2 <- 
function (x, width = getOption("width")) 
{
    if (is.expression(x) || is.list(x)) {
        trees <- vapply(x, tree2, character(1), width = width)
        out <- str_c(trees, collapse = "nn")
    }
    else {
        out <- tree2(x, width = width)
    }
    cat(out, "n")
}
environment(call_tree2)<-environment(pryr::call_tree)

最新更新