r - 在 R 表达式中查找所有函数的名称

Question

我正在尝试查找在任意合法 R 表达式中使用的所有函数的名称，但我找不到将以下示例标记为函数而不是名称的函数。

test <- expression(
    this_is_a_function <- function(var1, var2){

    this_is_a_function(var1-1, var2)
})

all.vars(test, functions = FALSE)

[1] "this_is_a_function" "var1"              "var2" 

all.vars(expr, functions = FALSE) 似乎在表达式中返回函数声明 (f <- function(){})，同时过滤掉函数调用 ('+'(1,2), ...)。

是否有任何函数 - 在核心库或其他地方 - 会将“this_is_a_function”标记为函数，而不是名称？它需要处理语法上合法但可能无法正确评估的任意表达式(例如'+'(1，'duck'))

我找到了 similar questions ，但它们似乎不包含解决方案。

如果需要澄清，请在下面发表评论。我正在使用解析器包来解析表达式。

编辑:@Hadley

我有包含整个脚本的表达式，这些脚本通常由一个包含嵌套函数定义的 main 函数组成，并在脚本末尾调用 main 函数。

函数都是在表达式中定义的，我不介意是否必须包含 '<-' 和 '{'，因为我可以自己轻松地将它们过滤掉。

动机是获取我所有的 R 脚本并收集有关我对函数的使用如何随时间变化的基本统计数据。

编辑:当前解决方案

基于正则表达式的方法抓取函数定义，结合 James 注释中的方法抓取函数调用。通常有效，因为我从不使用右手赋值。

function_usage <- function(code_string){
    # takes a script, extracts function definitions

    require(stringr)

    code_string <- str_replace(code_string, 'expression\\(', '')

    equal_assign <- '.+[ \n]+<-[ \n]+function'
    arrow_assign <- '.+[ \n]+=[ \n]+function'

    function_names <- sapply(
        strsplit(
            str_match(code_string, equal_assign), split = '[ \n]+<-'),    
        function(x) x[1])

    function_names <- c(function_names, sapply(
        strsplit(
            str_match(code_string, arrow_assign), split = '[ \n]+='),    
            function(x) x[1]))

        return(table(function_names))    
    }

Answer 1

简答: is.function检查变量是否实际包含一个函数。这不适用于(未评估的)调用，因为它们是调用。您还需要注意掩蔽:

mean <- mean (x)

更长的答案:

恕我直言， this_is_a_function 的两次出现之间存在很大差异。 .

在第一种情况下，您将为名为 this_is_a_function 的变量分配一个函数。一旦你评估表达式。区别与 2+2之间的区别相同和 4 .
然而，刚刚找到 <- function ()不保证结果是一个函数:

f <- function (x) {x + 1} (2)

第二次出现在语法上是一个函数调用。您可以从表达式中确定一个名为 this_is_a_function 的变量需要存在一个函数才能正确评估调用。但是:您不知道仅从该声明是否存在。但是，您可以检查这样的变量是否存在，以及它是否是一个函数。

事实上，函数也像其他类型的数据一样存储在变量中，这意味着在第一种情况下，您可以知道 function () 的结果。将是函数，并由此得出结论，在计算此表达式之后，名称为 this_is_a_function 的变量立即生效。将举行一个功能。

然而，R 充满了名称和函数:“->”是赋值函数的名称(一个保存赋值函数的变量)......

评估表达式后，您可以通过 is.function (this_is_a_function) 验证这一点。 .
然而，这绝不是返回函数的唯一表达式:想想

f <- function () {g <- function (){}}
> body (f)[[2]][[3]]
function() {
}
> class (body (f)[[2]][[3]])
[1] "call"
> class (eval (body (f)[[2]][[3]]))
[1] "function"

all.vars(expr, functions = FALSE) seems to return functions declarations (f <- function(){}) in the expression, while filtering out function calls ('+'(1,2), ...).

我想说的是反过来:在那个表达式中 f是将分配给函数的变量(名称)(一旦调用被评估)。 + (1, 2) 计算结果为数字。除非你阻止它这样做。

e <- expression (1 + 2)
> e <- expression (1 + 2)
> e [[1]]
1 + 2
> e [[1]][[1]]
`+`
> class (e [[1]][[1]])
[1] "name"
> eval (e [[1]][[1]])
function (e1, e2)  .Primitive("+")
> class (eval (e [[1]][[1]]))
[1] "function"