我正在尝试使用动态创建的名称创建一个新列,并使用涉及其他动态创建的变量的表达式填充该字段。例如,考虑以下数据框架。
ID multiplier value1_2015 value2_2015 value1_2016 value2_2016
1 0.5 2 3 1 4
2 1.0 2 4 5 1
我想编写一个给定数据框和一年的函数,然后仅计算相应年份变量的表达式,并将结果存储在名为total_year
的列中,其中{{1}是赋给函数的值。例如,如果表达式是
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
当我尝试这个时,我得到以下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)
}
使用Error in mutate_impl(.data, dots) :
Evaluation error: non-numeric argument to binary operator.
之类的内容替换表达式而不会出现错误,会生成正确的列名,但列中的值是字符串“value1_2016”,而不是名为{{1的列中的相应值}}。
答案 0 :(得分:8)
在这里,我们不需要enquo/quo_name
作为'year',因为我们传递的是数值。 paste
的输出将为character
类,使用sym
中的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