使用rlang double Curly Bracs {{in Data.Table

Question

问题

{{}}从rlang软件包运算符，使得将列名称作为函数参数（又称quasiquotation）非常容易。我了解rlang旨在与didyverse一起使用，但是有一种方法可以使用{{}} in data.table /代码>？

{{}}与Dplyr

test_dplyr <- function(dt, col1, col2){
  
  temp <- dt %>%
            group_by( {{col2}} ) %>%
            summarise(test = mean( {{col1}} ))

  return(temp)
}

test_dplyr(dt=iris, col1=Sepal.Length, col2=Species)

> # A tibble: 3 x 2
>   Species     test
>   <fct>      <dbl>
> 1 setosa      5.01
> 2 versicolor  5.94
> 3 virginica   6.59

失败的尝试使用{{}}的尝试使用{{}}与data.table一起

使用。这是我想做的，但它返回错误。

test_dt2 <- function(dt, col1, col2){
  
  data.table::setDT(dt)
  temp <- dt[, .( test = mean({{col1}})), by = {{col2}} ] )
  return(temp)
}

# error
test_dt2(dt=iris, col1= Sepal.Length, col2= Species)

# and error
test_dt2(dt=iris, col1= 'Sepal.Length', col2= 'Species')

rlang与data.table的替代用途

，这是使用rlang与data.table使用rlang的替代方法。这里有两个不一致的范围，这些不一致是rlang :: ensym（）每个列名称变量，并且必须调用data。

test_dt <- function(dt, col1, col2){
  
  # eval colnames
  col1 <- rlang::ensym(col1)
  col2 <- rlang::ensym(col2)
  
  data.table::setDT(dt)
  temp <- rlang::inject( dt[, .( test = mean(!!col1)), by = !!col2] )
  return(temp)
}

test_dt(dt=iris, col1='Sepal.Length', col2='Species')

>       Species  test
> 1:     setosa 5.006
> 2: versicolor 5.936
> 3:  virginica 6.588

Answer 1

我认为您不想将rlang与data.table一起使用。 Data.Table本身已经具有更方便的设施。还建议不要在此处使用setDT，因为这将导致更改DT的副作用。

library(data.table)

test_dt <- function(dt, col1, col2) {
  as.data.table(dt)[, .( test = mean(.SD[[col1]])), by = c(col2)]
}

test_dt(dt = iris, col1 = 'Sepal.Length', col2 = 'Species')
##       Species  test
## 1:     setosa 5.006
## 2: versicolor 5.936
## 3:  virginica 6.588

这也有效：

test_dt <- function(dt, col1, col2) {
  as.data.table(dt)[, .( test = mean(get(col1))), by = c(col2)]
}

test_dt(dt=iris, col1='Sepal.Length', col2='Species')