data.table的包装函数在本地环境中不起作用

时间:2014-08-20 23:42:16

标签: r data.table environment subset

考虑以下功能:

test <- function(x,...) {
  # in practical case, it does more
  x[...]
}

然后调用子集的包装函数在本地环境中不起作用。

> library(data.table)
> m <- data.table(x=1:3,key="x")
> m[J(1)]
   x
1: 1
> local({i <- 1; m[J(i)]})
   x
1: 1
> local({i <- 1; test(m, J(i))})
Error in eval(expr, envir, enclos) : object 'i' not found
> local({i <- 1; test(m, i)})
Error in eval(expr, envir, enclos) : object 'i' not found

可能是因为test()没有尝试在该环境中查找符号。但是,如果我将data.table更改为data.frame并运行最后一行,则可以正常工作。

> m <- data.frame(x=1:3)
> local({i <- 1; test(m, i)})
  x
1 1
2 2
3 3

如何修改test()以便它可以与...一起使用并为data.table进行子集化?

我的会话信息:

> sessionInfo()
R version 3.1.1 (2014-07-10)
Platform: x86_64-w64-mingw32/x64 (64-bit)

locale:
[1] LC_COLLATE=English_United States.1252  LC_CTYPE=English_United States.1252   
[3] LC_MONETARY=English_United States.1252 LC_NUMERIC=C                          
[5] LC_TIME=English_United States.1252    

attached base packages:
[1] stats     graphics  grDevices utils     datasets  methods   base     

other attached packages:
[1] data.table_1.9.2

loaded via a namespace (and not attached):
[1] plyr_1.8.1    Rcpp_0.11.2   reshape2_1.4  stringr_0.6.2 tools_3.1.1  

1 个答案:

答案 0 :(得分:5)

因为data.table的[...]采用在parent.frame中计算的表达式,所以它不起作用是有意义的。因此,在这种情况下,J(i)将在test()的环境中进行评估,因为它会被调用。test()由于test()是在全局环境中定义的,因此默认情况下将在全局环境中搜索test()中未找到的任何变量。 test()功能与您创建的本地环境并不真正相关。您可以更改要在其父级框架中评估的test <- function(x,...) { cc <- match.call() cc[[1]] <- quote(`[`) names(cc)[2]<-"" #make data.frame happy eval.parent(cc) } 函数。例如

library(data.table)
m <- data.table(x=1:3,key="x")    
n <- data.frame(x=1:3)

测试数据

local({i <- 1; m[J(i)]})
#    x
# 1: 1

local({i <- 1; test(m, J(i))})
#    x
# 1: 1

local({i <- 1; test(n, i)})
#   x
# 1 1
# 2 2
# 3 3

i<-2
test(m, J(i))
#    x
# 1: 2

现在功能

{{1}}