用tidyeval在map2中取消引用

时间:2017-08-02 15:15:01

标签: r rlang tidyeval

我正在创建一个计算“运行”次数或缺失或完整数据的函数 - 我希望这与dplyr::group_by一起使用,所以我把它写成S3方法 - 下面是一个简化的例子这段代码。

不幸的是,我发现裸露的未加引号的变量名称不起作用,但引用它,这确实有效,奇怪的是。

以下是输出

的示例
  fun_run <- function(data, var) {

    UseMethod("fun_run")

  }

fun_run.default <- function(data, var) {

  var <- rlang::enquo(var)

  data_pull <- data %>% dplyr::pull(!(!var))

  # find the lengths of the number of missings in a row
  tibble::as_tibble(c(rle(is.na(data_pull))))

}

fun_run.grouped_df <- function(data, var) {

  var <- rlang::enquo(var)

  tidyr::nest(data) %>% dplyr::mutate(data = purrr::map2(.x = data, .y = !(!var), 
                                                         .f = fun_run)) %>% tidyr::unnest()

}

library(dplyr)
#> 
#> Attaching package: 'dplyr'
#> The following objects are masked from 'package:stats':
#> 
#>     filter, lag
#> The following objects are masked from 'package:base':
#> 
#>     intersect, setdiff, setequal, union

airquality %>% fun_run(Ozone)
#> # A tibble: 35 x 2
#> lengths values
#> <int>  <lgl>
#>   4    FALSE
#>   1     TRUE
#>   4    FALSE
#>   1     TRUE
#>  14    FALSE
#>   3     TRUE
#>   4    FALSE
#>   6     TRUE
#>   1    FALSE
#>   1     TRUE
#> ... with 25 more rows

# doesn't work
airquality %>% group_by(Month) %>% fun_run(Ozone)
#> Error in mutate_impl(.data, dots) : Evaluation error: object 'Ozone' not found. 

# does work
airquality %>% group_by(Month) %>% fun_run("Ozone")
#> # A tibble: 37 x 3
#> Month lengths values
#> <int>   <int>  <lgl>
#>     5       4  FALSE
#>     5       1   TRUE
#>     5       4  FALSE
#>     5       1   TRUE
#>     5      14  FALSE
#>     5       3   TRUE
#>     5       4  FALSE
#>     6       6   TRUE
#>     6       1  FALSE
#>     6       1   TRUE
#> # ... with 27 more rows

1 个答案:

答案 0 :(得分:2)

您实际上并不想使用map2,因为您的第二个输入(var)不会随着第一个输入(分组/嵌套{{1})而改变})。此外,&#34;臭氧&#34;列在此时隐藏在嵌套数据中。你可以通过尝试执行没有任何tidyeval语法的代码来看到这一点:

data

相反,您希望使用标准data <- airquality %>% group_by(Month) tidyr::nest(data) %>% dplyr::mutate(data = purrr::map2(.x = data, .y = Ozone, .f = fun_run)) %>% tidyr::unnest() #>Error in mutate_impl(.data, dots) : #> Evaluation error: object 'Ozone' not found.

map

重写以便在您的函数中使用:

tidyr::nest(data) %>% dplyr::mutate(data = purrr::map(.x = data, var = Ozone, 
                                                       .f = fun_run)) %>% tidyr::unnest()

这会产生最终引用示例的结果。