我正在尝试编写一个函数,该函数可以将函数作为 Rcpp 中的参数。我在 R 中编写了一个示例函数,它显示了我想要的功能类型:
simulate_and_evaluate <- function(simulate, evaluate) {
y <- simulate(1)
eval <- evaluate(y)
return(eval)
}
simulate_fun <- function(n) rnorm(n, 0, 1)
evaluate_fun <- function(x) dnorm(x, 0, 1)
simulate_and_evaluate(simulate = simulate_fun,
evaluate = evaluate_fun)
Run Code Online (Sandbox Code Playgroud)
在这个函数中simulate_and_evaluate,它接受两个都是函数的参数,一个模拟一个数字,一个用这个模拟数字计算一个函数。例如,我们可以从标准法线模拟一个值,并评估该点标准法线的密度。有谁知道在 Rcpp 中是否有办法做到这一点?
Rcpp 旨在实现R 和 C++ 对象的无缝接口。由于函数是在内部表示为 aSEXP可以采用的类型的第一类 R 对象,因此我们当然也可以将它们与 Rcpp 一起发送。有很多例子。
所以在这里我们简单地将您的函数重写为 C++ 函数:
Rcpp::cppFunction("double simAndEval(Function sim, Function eval) {
double y = as<double>(sim(1));
double ev = as<double>(eval(y));
return(ev);
}")
Run Code Online (Sandbox Code Playgroud)
然后我们可以将 RNG 设置为相同的值,运行您的 R 函数和这个 C++ 函数并获得相同的值。这太棒了。
R> set.seed(123)
R> simulate_and_evaluate(simulate = simulate_fun,
+ evaluate = evaluate_fun)
[1] 0.341
R> set.seed(123) # reset RNG
R> simAndEval(simulate_fun, evaluate_fun)
[1] 0.341
R>
Run Code Online (Sandbox Code Playgroud)
但是正如@MrFlick 警告你的那样,这不会运行得更快,因为我们没有添加实际函数的编译执行,我们只是从 C++ 而不是 R 调用它们。
这个话题之前已经讨论过了。请搜索 StackOverflow,也许用一个字符串[rcpp] Function来获得一些有意义的点击。