R:on.exit - 使用返回值而不知道其名称

时间:2015-03-25 17:32:17

标签: r data.table devtools

我有以下功能。除了函数中的第一个代码块之外,我无法以任何方式更改函数 在这个简单的例子中,我想显示在返回对象上应用一些函数 关键是函数返回的变量名称可能会有所不同,我无法猜测它 显然,我也无法将f函数包装到{ x <- f(); myfun(x); x } 我的.Last.value调用中的以下on.exit表示f函数返回的值。

f <- function(param){
  # the only code I know - start
  on.exit(if("character" %in% class(.Last.value)) message(print(.Last.value)) else message(class(.Last.value)))
  # the only code I know - end

  # real processing of f()

  a <- "aaa"
  "somethiiiing"
  if(param==1L) return(a)

  b <- 5L
  "somethiiiing"
  if(param==2L) return(b)

  "somethiiiing"
  return(32)
}
f(1L)
# function
# [1] "aaa"
f(2L)
# aaa
# [1] 5
f(3L)
# integer
# [1] 32

以上代码.Last.value似乎正在使用延迟(所以实际上不起作用)而且.Last.value可能不是我想要的方式使用像if(fun0(x)) fun1(x) else fun2(x)这样的值几次,并且因为返回的值可能是一个大对象,所以复制它的方法也是不好的方法。
有什么方法可以使用on.exit或任何其他函数来帮助我在f函数结果上运行我的函数而不知道结果变量名吗?

2 个答案:

答案 0 :(得分:1)

与修改函数的方式类似,您也可以轻松地将其包装起来。这是一个可重复的例子。

library(data.table)

append.log<-function(x) {
    cat(paste("value:",x,"\n"))
}

idx.dt <- data.table:::`[.data.table`
environment(idx.dt)<-asNamespace("data.table")

idx.wrap <- function(...) {
    x<-do.call(idx.dt, as.list(substitute(...())), envir=parent.frame())
      append.log(if(is(x, "data.table")) {
          nrow(x)
      } else { NA })
      x
}
environment(idx.wrap)<-asNamespace("data.table")

(unlockBinding)("[.data.table",asNamespace("data.table"))
assign("[.data.table",idx.wrap,envir=asNamespace("data.table"),inherits=FALSE)


dt<-data.table(a=1:10, b=seq(2, 20, by=2), c=letters[1:10])

dt[a%%2==0]

答案 1 :(得分:1)

由于新功能returnValue,完全有可能是R 3.2.0。 下面的工作示例。

f <- function(x, err = FALSE){
  pt <- proc.time()[[3L]]
  on.exit(message(paste("proc.time:",round(proc.time()[[3L]]-pt,4),"\nnrow:",as.integer(nrow(returnValue()))[1L])))
  Sys.sleep(0.001)
  if(err) stop("some error")
  return(x)
}
dt <- data.frame(a = 1:5, b = letters[1:5])
f(dt)
f(dt, err=T)
f(dt)
f(dt[dt$a %in% 2:3 & dt$b %in% c("c","d"),])