r - 使用 dplyr 非标准评估创建多个函数

Question

我正在尝试使用 mutate_() 创建多个列，其中每个列都基于使用不同输入调用的自定义函数。我可以使用 paste() 创建多个带引号的函数调用，但这不起作用，因为 dplyr 的 NSE 需要公式 (~) 而不是带引号的字符串才能找到函数。如何在下面写“dots =”行以便可以找到该函数？我尝试使用 ~、as.formula() 和lazyeval::interp() 进行试验，但都无法正常工作。我的实际“前缀”是一个长向量，所以我不想单独写出每个新列的函数调用。谢谢

library(dplyr)
library(lazyeval)
library(nycflights13)

myfunc = function(x, y) { x - y }

# this works
flights1 <- mutate(flights, dep_time_sched = myfunc(dep_time, dep_delay), 
                            arr_time_sched = myfunc(arr_time, arr_delay))

# this doesn't - Error: could not find function "myfunc"
prefixes <- c('dep', 'arr')
dots = as.list(paste0('myfunc(', 
                       paste0(prefixes, '_time'), ', ', 
                       paste0(prefixes, '_delay)')))
flights2 <- mutate_(flights, .dots = setNames(dots, paste0(prefixes, '_time_sched')))

Answer 1

您可以通过使用 interp 和 lapply 来循环访问您的前缀并获取 mutate_ 所需格式的列表。

    dots = lapply(prefixes, function(var) interp(~myfunc(x, y), 
                                        .values = list(x = as.name(paste0(var, "_time")),
                                                    y = as.name(paste0(var, "_delay")))))
    dots

[[1]]
~myfunc(dep_time, dep_delay)
<environment: 0x0000000019e51f00>

[[2]]
~myfunc(arr_time, arr_delay)
<environment: 0x0000000019f1e5b0>

这给出了与您的 flights1 相同的结果。

flights2 = mutate_(flights, .dots = setNames(dots, paste0(prefixes, '_time_sched')))
identical(flights1, flights2)
[1] TRUE