有条件地使用 is.null() 子集 r data.table

Question

我有一个data.table

library(data.table)

testDT <- data.table(
        L = (1:32), 
        M = rep(letters[23:26], each = 64), 
        N = rep(LETTERS[1:2], times = 2, each = 512),
        O = rnorm(2048, 1))

testDT$L                <- factor(testDT$L,         levels = seq(from = 1, to = 32, by = 1))

我创建了一个函数来有条件地对数据集进行子集化。如果子集变量G 是NULL 并且H 是"w"，那么我希望testDT$N 中的所有值和testDT$M 中的所有值"w" 都在testDT 中返回。这是我创建的，但无法正常运行：

G <- NULL 
H <- "w"

testDT1 <- testDT[if(is.null(G)) {eval(call("%in%", as.name("N"), G))} & 
                if(is.null(H)) {eval(call("%in%", as.name("M"), H))}]

我验证了除了 if(is.null()) 部分之外的所有内容都可以通过创建它来正常工作，它的子集正确：

G <- "A" 
H <- "w"
testDT1 <- testDT[{eval(call("%in%", as.name("N"), G))} & 
                {eval(call("%in%", as.name("M"), H))}]

如何正确使用is.null() 条件？

Answer 1

使用语言计算，您可以使用专用函数准备call 对象。

library(data.table)
testDT = data.table(
    L = factor(1:32), 
    M = rep(letters[23:26], each = 64), 
    N = rep(LETTERS[1:2], times = 2, each = 512),
    O = rnorm(2048, 1)
)

i.expr = function(var, x){
    if(is.null(x)) TRUE
    else call("%in%", as.name(var), x)
}

G = NULL
H = "w"
i.expr("N",G)
#[1] TRUE
i.expr("M",H)
#M %in% "w"
testDT1 = testDT[eval(i.expr("N",G)) & eval(i.expr("M",H))]

G = "A" 
H = "w"
i.expr("N",G)
#N %in% "A"
i.expr("M",H)
#M %in% "w"
testDT2 = testDT[eval(i.expr("N",G)) & eval(i.expr("M",H))]

如果您总是按两个条件和& 运算符作为子集。我会将它合并到一个函数中，这样你就可以使用testDT[eval(i.expr(...))] 调用它一次。