子集data.table按逻辑列

Question

我有一个带逻辑列的data.table 。 为什么逻辑列的名称不能直接用于i参数？ 参见示例。

dt <- data.table(x = c(T, T, F, T), y = 1:4)

# Works
dt[dt$x]
dt[!dt$x]

# Works
dt[x == T]
dt[x == F]

# Does not work
dt[x]
dt[!x]

Answer 1

从?data.table

高级：当i是单个变量名时，它不被视为列名的表达式，而是在调用范围内求值。

因此dt[x]将尝试在调用范围内评估x （在这种情况下为全局环境）

您可以使用(或{或force

dt[(x)]
dt[{x}]
dt[force(x)]

Answer 2

x在全局环境中未定义。 如果你尝试这个

> with(dt, dt[x])
      x y
1: TRUE 1
2: TRUE 2
3: TRUE 4

会的。 或这个：

> attach(dt)
> dt[!x]
       x y
1: FALSE 3

编辑：

根据文档， j参数采用列名，实际上：

> dt[x]
Error in eval(expr, envir, enclos) : object 'x' not found
> dt[j = x]
[1]  TRUE  TRUE FALSE  TRUE

然后， i参数采用数值或逻辑表达式（例如x本身应该是），但是似乎（data.table）如果没有以下内容，则无法将x视为逻辑x ：

> dt[i = x]
Error in eval(expr, envir, enclos) : object 'x' not found
> dt[i = as.logical(x)]
      x y
1: TRUE 1
2: TRUE 2
3: TRUE 4

Answer 3

这也应该起作用，并且可以说是更自然的：

setkey(dt, x)
dt[J(TRUE)]
dt[J(FALSE)]