动态构建查询多列的调用

时间:2015-05-26 20:39:19

标签: r dynamic data.table lookup

如何使用字符向量变量作为参数动态查找多个字段并通过引用添加。在下面的情况下,我想查找两列并删除其中的i.前缀。当然,他们可以覆盖已经存在的具有相同名称的列。

library(data.table)
set.seed(1)
ID <- data.table(id = 1:3, meta = rep(1,3), key = "id")
JN <- data.table(idd = sample(ID$id, 3, FALSE), value = sample(letters, 3, FALSE), meta = rep(1,3), key = "idd")
select <- c("value","meta") # my fields to lookup
j.lkp <- call(":=", select, lapply(paste0("i.",select), as.symbol))
j.lkp
# `:=`(c("value", "meta"), list(i.value, i.meta))
ID[JN, eval(j.lkp)]
# Error in eval(expr, envir, enclos) : could not find function "i.value"
ID[JN, `:=`(c("value", "meta"), list(i.value, i.meta))]
#    id meta value
# 1:  1    1     x
# 2:  2    1     v
# 3:  3    1     f

我知道similar question,但是这个请求在加入过程中要求矢量化参数并直接构建j的调用。
编辑:我知道我可以使用.SDcols执行此操作,但之后我无法通过引用执行此

3 个答案:

答案 0 :(得分:9)

这对我来说似乎是最直截了当的方式:

ID[JN, (select) := mget(paste0('i.', select))]

答案 1 :(得分:5)

这是粗暴的方式:

myj <- parse(text=paste0("`:=`(",paste0(select,"=i.",select,collapse=","),")"))
ID[JN,eval(myj)]
#    id meta value
# 1:  1    1     x
# 2:  2    1     v
# 3:  3    1     f

答案 2 :(得分:5)

代替mgeteval-parse,仍有可能构建查询调用。虽然mget是最友好的用户,但这个版本既灵活又实际上对应于构建j表达式。
解包装到batch.lookup辅助函数中,使用列名的字符向量进行查找。

library(data.table)
set.seed(1)
ID <- data.table(id = 1:3, meta = rep(1,3), key = "id")
JN <- data.table(idd = sample(ID$id, 3, FALSE), value = sample(letters, 3, FALSE), meta = rep(1,3), key = "idd")
select <- c("value","meta") # my fields to lookup

batch.lookup = function(x) {
    as.call(list(
        as.name(":="),
        x,
        as.call(c(
            list(as.name("list")),
            sapply(x, function(x) as.name(paste0("i.",x)), simplify=FALSE)
        ))
    ))
}
batch.lookup(select)
#`:=`(c("value", "meta"), list(value = i.value, meta = i.meta))
ID[JN, eval(batch.lookup(select))][]
#   id meta value
#1:  1    1     x
#2:  2    1     v
#3:  3    1     f

公平地说,这个答案实际上解决了我所描述的呼叫建设问题OP。