R 使用 ：= 将 data.table 列名传递给函数

set是数据.餐桌习惯用语.如果你需要做其他事情，比如use by，rlang有一种延迟计算的通用方法，那就是enexpr args(或者enquo，如果你想在原始环境中对它们进行计算)，然后用你通常使用的表达式将它们放入inject中.

library(rlang)
#> Warning: package 'rlang' was built under R version 4.1.2
library(data.table)
#> 
#> Attaching package: 'data.table'
#> The following object is masked from 'package:rlang':
#> 
#>     :=

dat <- data.table(x = 1:4,
                  y = 5:8)

new_column <- function(df, col_name, expr) {
    col_name <- enexpr(col_name)
    expr <- enexpr(expr)
    inject(df[, !!col_name := !!expr])
}

new_column(dat, z, x + y)

dat
#>        x     y     z
#>    <int> <int> <int>
#> 1:     1     5     6
#> 2:     2     6     8
#> 3:     3     7    10
#> 4:     4     8    12

^{由reprex package(v2.0.1)于2022年2月25日创建}

或者，同样没有rlang

library(data.table)

dat <- data.table(x = 1:4,
                  y = 5:8)

new_column <- function(df, col_name, expr) {
    col_name <- deparse(substitute(col_name))
    expr <- substitute(expr)
    df[, (col_name) := eval(expr)]
}

new_column(dat, z, x + y)

dat
#>        x     y     z
#>    <int> <int> <int>
#> 1:     1     5     6
#> 2:     2     6     8
#> 3:     3     7    10
#> 4:     4     8    12

^{由reprex package(v2.0.1)于2022年2月25日创建}

数据.在开发版本中，table包还有一个新的"语言编程"接口，但我能说的最好的情况是，它只允许符号，不允许表达式.

https://rdatatable.gitlab.io/data.table/news/index.html