r - 从 data.table 中的 `by` 内访问 `j` 变量

Question

如果我有一个 data.table 并且我正在 j 中执行一些函数调用，我是否可以访问 by 变量的当前值？

library(data.table)
d <- data.table(x=1:10, y=c('a', 'b'))

myfun <- function(DT) {
  print (DT$y)
}


d[, myfun(.SD), by=y]

有关更多上下文，我将第二个参数(另一个 data.table )传递给 myfun 并希望根据 y 的当前值对其进行子集化。它可以用一个虚拟变量来完成，但这看起来很糟糕......

Answer 1

使用 .BY - 这是 by 变量的列表:

d <- data.table(x=1:10, y=c('a', 'b'))
d[, .BY[[1]], by = y]  # [[1]] to access the first by variable, which is y
                       # if you had by = list(x, y) you'd do .BY[[2]] to access y
#   y V1
#1: a  a
#2: b  b

此外，该列表已命名，因此您还可以按名称访问它:

d[, .BY$y, by = y]