奇怪的 dplyr+NSE：在 mutate 的自定义函数中使用 NSE

Question

在dplyr::mutate 内的自定义函数中使用 NSE 时，我偶然发现了一个问题。考虑以下代码：

require(tidyverse)

f <- function(var) {
  varname <- deparse(substitute(var))
  v1      <- as.name(sprintf("%s.Width", varname))
  v2      <- as.name(sprintf("%s.Length", varname))
  return(substitute(v1 + v2))
}

iris %>% 
  mutate(
    test = f(Sepal) %>% eval()
  )

ff <- function(var) {
  varname <- deparse(substitute(var))
  v1      <- as.name(sprintf("%s.Width", varname))
  v2      <- as.name(sprintf("%s.Length", varname))
  substitute(v1 + v2) %>% eval.parent(n = 1)
}

iris %>% 
  mutate(
    test = ff(Sepal)
  )

这里f 工作正常，但需要对eval() 进行外部调用才能在mutate() 环境中执行代码。 这当然有点难看，并导致大量样板代码。我对这项工作的最佳猜测是ff，它试图在其调用环境中评估构造的表达式——我预计它是mutate() 环境。但是，这会引发 variable-not-found 错误。 关于如何进行这项工作以及潜在问题是什么的任何想法？本质上，我希望在 dplyr 动词中允许自定义“宏”。

Answer 1

使用!! tidy eval "unquote" 运算符：

ff <- function(var) {
  varname <- deparse(substitute(var))
  v1      <- as.name(sprintf("%s.Width", varname))
  v2      <- as.name(sprintf("%s.Length", varname))
  substitute(v1 + v2)
}

iris %>% 
  head() %>% 
  mutate(
    test = !! ff(Sepal)
  )