传入函数作为输入和返回函数

Question

我想编写一个 R 函数，它采用 x 中的数学函数并在 x 中返回一个新函数作为输出。例如：

输入应作为数学函数（或关系）传入x：

g <- x^2 + 9*x + log(x)

结果输出应该是：

function(x) (exp(g))

即我想在这个说明性示例中返回 x 中原始函数的符号指数表达式，即 exp(x^2 + 9*x + log(x))

所以理想情况下它会返回函数对象：

function(x) (exp(x^2 + 9*x + log(x)))

我尝试如下：

test <- function(g){
h <- function(x){exp(g)}
return(h)
}
m <- test(x^2 + 9*x + log(x))
m(10)

所以 m(10) 应该返回：

exp(10^2 + 9*10 + log(10))

在这种情况下是 exp(192.3026)。

谁能告诉我如何做到这一点？

Answer 1

你可以使用包函数：

library(functional)
fun <- Compose(function(x) x^2 + 9*x + log(x), exp)
fun(1)
#[1] 22026.47

Answer 2

这是一种方法：

test <- function(e) {
    ee <- substitute(e)
    eee <- substitute(exp(X), list(X=ee))
    f <- function(x) {}
    body(f) <- eee
    environment(f) <- parent.frame()
    f
}

## Check that it works
m <- test(x^2 + 9*x + log(x))
m
# function (x) 
# exp(x^2 + 9 * x + log(x))
m(1)
# [1] 22026.47
m(1) == exp(10)
# [1] TRUE

Answer 3

编辑 - 有问题的功能

f <- function(...) {
  l <- eval(substitute(alist(x = x, ...)))
  l[[2]] <- substitute(exp(X), list(X = l[[2]]))
  as.function(`names<-`(l, l[sapply(l, is.symbol)]))
}

g <- f(x^2 + 2*x + 5)
# function (x = x) 
#   exp(x^2 + 2 * x + 5)

g(1)
# [1] 2980.958

---

这是一般情况的另一种方式：

f <- function(...) {
  l <- eval(substitute(alist(...)))
  as.function(`names<-`(l, l[sapply(l, is.symbol)]))
}

g <- f(x, x^2 + 9*x + log(x))

# function (x = x) 
#   x^2 + 9 * x + log(x)

g(10)
# [1] 192.3026

此版本也适用于任意数量的变量，只需定义它们后跟函数：

g <- f(x, y, z, x + 2 * y + z ** 3)

# function (x = x, y = y, z = z) 
#   x + 2 * y + z^3

g(1, 2, 0)
# [1] 5

---

可能有更好的方法将... 添加到函数中，但您可以这样做

f <- function(..., use_dots = FALSE) {
  l <- eval(substitute(alist(...)))
  if (use_dots)
    l <- c(head(l, -1), list('...' = as.symbol('...')), tail(l, 1))
  as.function(`names<-`(l, l[sapply(l, is.symbol)]))
}

所以现在您不必命名所有变量/参数

g <- f(x, y, plot(x, y, ...), use_dots = TRUE)
g(1:5, 1:5, main = 'main title', pch = 16, col = 3, cex = 3, xpd = NA)