r - 将别名列传递给函数以用于 data.table `by=` 参数

Question

我希望能够对按不同属性分组的 data.table 进行大量批量处理。

我有一个函数，它允许通过将要分组的列的字符串传递给它来完成此操作:

library(data.table)

myaggfunction<-function(mydt,aggcol,type){
  mydt[,lapply(.SD, sum), by=aggcol][,Type:=type]
}

myaggfunction(as.data.table(iris),"Species","Test")[]

但是，某些列需要在输出中使用别名。我知道我可以将字符串传递给 by但是如何传递不在函数参数级别进行评估的等效列表，例如

myaggfunction(as.data.table(iris),list(IrisSpecies=Species),"Test")[]

Answer 1

您可以使用 substitute获取未计算的表达式并将其传递给 by在 data.table .

library(data.table)

myaggfunction<-function(mydt,aggcol,type){
  aggcol <- substitute(aggcol)
  mydt[,lapply(.SD, sum), by=aggcol][,Type:=type]
}

myaggfunction(as.data.table(iris),"Species","Test")[]
myaggfunction(as.data.table(iris),list(IrisSpecies=Species),"Test")[]