如何在R-dplyr函数中引用用:=创建的动态列？

问题原因 #

在同一mutate()调用中，直接用{{col}}_d这种语法引用刚创建的动态列会触发"非数值参数"错误，核心原因是{{}}在函数调用阶段就会捕获输入表达式，而:=创建的列是在mutate执行阶段才生成，两者解析时机不匹配，导致R无法识别{{col}}_d对应的列对象。

解决方案 #

以下三种方法均可解决该问题：

方法1：使用.data代词结合动态列名 #

先通过as_label(enquo(col))获取输入列的字符串名称，拼接出新列名后，用.data[[ ]]引用刚创建的列：

library(dplyr)

my_func <- function(df, col) {
  col_str <- as_label(enquo(col))
  df %>%
    mutate(
      {{col}}_d := {{col}} * 2,
      {{col}}_e := .data[[paste0(col_str, "_d")]] + 3
    )
}

# 测试
test_df <- tibble(x = 1:5)
my_func(test_df, x)

方法2：用!!sym()动态解析列名 #

先构造新列的符号对象，再用!!强制解析引用该列：

my_func <- function(df, col) {
  col_str <- as_label(enquo(col))
  col_d_sym <- sym(paste0(col_str, "_d"))
  df %>%
    mutate(
      {{col}}_d := {{col}} * 2,
      {{col}}_e := !!col_d_sym + 3
    )
}

方法3：拆分mutate调用 #

将列的创建和引用拆分为两次mutate操作，第二次调用时已生成的动态列可直接被识别：

my_func <- function(df, col) {
  df %>%
    mutate({{col}}_d := {{col}} * 2) %>%
    mutate({{col}}_e := {{col}}_d + 3)
}

测试结果 #

三种方法运行后都会输出预期结果：

# A tibble: 5 × 3
      x   x_d   x_e
  <int> <dbl> <dbl>
1     1     2     5
2     2     4     7
3     3     6     9
4     4     8    11
5     5    10    13

内容的提问来源于stack exchange，提问作者kmlewis15