从评估函数返回 system.time

Question

R 版本 2.12，Windows XP

我正在尝试编写一个函数（如“g”），它接受一个参数、一个函数（如“f”），并返回匹配的函数。此外，“g”主体中包含一条语句，该语句告诉结果对象在调用该对象时返回 system.time 的值。举个例子就可以清楚地说明这一点。

我想要什么：

g <- function(f) {...}
z <- g(mean)
z(c(1, 4, 7))

带输出

user system elapsed
0.04   0.00    0.04

我拥有什么：

g <- function(f) {if (!exists("x")) {x <- match.fun(f)} else {system.time(x)}}
z <- g(mean)
z(c(1, 4, 7))

带输出

[1] 4

非常感谢任何帮助。

Answer 1

也许这会有所帮助：

g <- function(f){
  function(x){
    zz <- system.time(
        xx <- match.fun(f)(x)
    )
    list(value=xx, system.time=zz)
  }
}

使用中：

g(mean)(c(1, 4, 7))

$value
[1] 4

$system.time
   user  system elapsed 
      0       0       0 

---

您可能需要考虑如何返回值。我使用了一个列表，但另一个选择是打印系统时间作为副作用并返回计算值。

Answer 2

最近我为自己做了类似的功能：

with_times <- function(f) {
    f <- match.fun(f)
    function(...) {
        .times <- system.time(res <- f(...))
        attr(res, "system.time") <- as.list(na.omit(.times))
        res
    }
}

例如：

g <- function(x,y) {r<-x+y; Sys.sleep(.5); r}
g(1, 1)
# [1] 2
g2 <- with_times(g) 
w <- g2(1, 1)

可以通过两种方式提取计时：

attributes(w)$system.time
# $user.self
# [1] 0
# $sys.self
# [1] 0
# $elapsed
# [1] 0.5

或

attr(w, "system.time")
# $user.self
# [1] 0
# $sys.self
# [1] 0
# $elapsed
# [1] 0.5