Dplyr - Mutate使用其他动态命名的变量动态命名变量

Arz*_*anK 10 r dplyr

我正在尝试使用动态创建的名称创建一个新列,并使用涉及其他动态创建的变量的表达式填充该字段.例如,考虑以下数据框架.

ID    multiplier    value1_2015    value2_2015    value1_2016    value2_2016
 1           0.5              2              3              1              4
 2           1.0              2              4              5              1
Run Code Online (Sandbox Code Playgroud)

我想编写一个给定数据框和一年的函数,然后只计算相应年份变量的表达式,并将结果存储在一个名为的列中total_year,其中year是给函数的值.例如,如果表达式是

multiplier * value1_year + value2_year我打电话给my_fun(df, 2016)我,我应该收到

ID multiplier value1_2015 value2_2015 value1_2016 value2_2016  total_2016
 1        0.5           2           3           1           4         4.5
 2        1.0           2           4           4           5           9
Run Code Online (Sandbox Code Playgroud)

这就是我所拥有的

my_fun <- function(df, year) {

 year <- enquo(year)

 total_header <- paste("total", quo_name(year), sep = "_")
 calc1_header <- paste("value1", quo_name(year), sep = "_")
 calc2_header <- paste("value2", quo_name(year), sep = "_")

 calc1_header <- enquo(calc1_header)
 calc2_header <- enquo(calc2_header)

 ret_table <- df %>%
 mutate(!!total_header := multiplier * !!calc1_header + !!calc2_header)

 return(ret_table)

}
Run Code Online (Sandbox Code Playgroud)

当我尝试这个时,我得到以下内容 Error in mutate_impl(.data, dots) : Evaluation error: non-numeric argument to binary operator.

将表达式替换为只!!total_header := !!calc1_header运行而没有错误,生成正确的列名,但列中的值是字符串"value1_2016",而不是名为列的相应值value1_2016.

akr*_*run 9

在这里,我们不需要enquo/quo_name'年',因为我们传递的是数值.输出paste将是characterclass,使用symfrom rlang(如提及@joran),这可以转换为符号并使用!!.确保在'!!周围添加牙箍!calc1_header'和'!! calc2_header'来评估特定对象

my_fun <- function(df, year) {

  total_header <- paste("total", year, sep = "_")
  calc1_header <- rlang::sym(paste("value1", year, sep = "_"))
  calc2_header <- rlang::sym(paste("value2", year, sep = "_"))

 df %>%
       mutate(!!total_header := multiplier * (!!calc1_header) + (!!calc2_header))



}

my_fun(df1, 2016)
#   ID multiplier value1_2015 value2_2015 value1_2016 value2_2016 total_2016
#1  1        0.5           2           3           1           4        4.5
#2  2        1.0           2           4           4           5        9.0
Run Code Online (Sandbox Code Playgroud)

  • 从 `rlang` v0.2.0 开始,不再需要 `!!var` 周围的大括号。 (2认同)