1

我想编写一个处理数据的自定义函数。函数的输入应该是：

  1. data.frame 对象
  2. 数据中与函数将要执行的操作相关的列的名称。

我想对这个函数进行编程，使指定列名的参数尽可能灵活。因此，我决定使用dot-dot-dot ( ...)。

但是，我不知道如何合并tidyselect的选择助手（即contains,starts_with等）

例子

例如，假设我想为dplyr::coalesce().

library(rlang)
library(dplyr)

my_coalesce_func <- function(dat, ...) {
  
  # step 1: save `...` into cols (https://tidyeval.tidyverse.org/multiple.html)
  cols <- enquos(...)
  
  # step 2: mutate new column `new_col` that coalesces columns of interest
  mutate(dat, new_col = coalesce(!!!cols))
}

# toy data
my_df <-
  data.frame(
    col_a = c(2, 2, 5, 5, 3),
    col_b = c(NA, 4, 2, 3, 1),
    col_c = c(4, 5, 3, 1, 2),
    col_d = c(1, NA, 4, 2, 4),
    col_e = c(3, 3, 1, 4, 5),
    extra_col = 1:5
  )

# run the function -- works fine when we explicitly provide column names 
my_coalesce_func(dat = my_df, col_a, col_b, col_c, col_d, col_e)
#>   col_a col_b col_c col_d col_e extra_col new_col
#> 1     2    NA     4     1     3         1       2
#> 2     2     4     5    NA     3         2       2
#> 3     5     2     3     4     1         3       5
#> 4     5     3     1     2     4         4       5
#> 5     3     1     2     4     5         5       3


# run the function -- fails to use a select helper
my_coalesce_func(dat = my_df, starts_with("col"))
#> Error: Problem with `mutate()` column `new_col`.
#> i `new_col = coalesce(starts_with("col"))`.
#> x `starts_with()` must be used within a *selecting* function.
#> i See <https://tidyselect.r-lib.org/reference/faq-selection-context.html>.

由reprex 包于 2021-07-01 创建 (v2.0.0 )

我需要添加什么才能my_coalesce_func()成功运行

my_coalesce_func(dat = my_df, starts_with("col"))

或传递给的任何其他选择助手...。

谢谢！

4

1 回答 1

1

基于这个答案，一个技巧是依靠dplyr::select()：

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

my_coalesce_func = function(data, ...) {
  data %>%
    select(...) %>%
    transmute(new_col = coalesce(!!!.)) %>% 
    bind_cols(data, .)
}

my_df <-
  data.frame(
    col_a = c(2, 2, 5, 5, 3),
    col_b = c(NA, 4, 2, 3, 1),
    col_c = c(4, 5, 3, 1, 2),
    col_d = c(1, NA, 4, 2, 4),
    col_e = c(3, 3, 1, 4, 5),
    extra_col = 1:5
  )

# explicitly provide column names
my_coalesce_func(dat = my_df, col_d, col_e)
#>   col_a col_b col_c col_d col_e extra_col new_col
#> 1     2    NA     4     1     3         1       1
#> 2     2     4     5    NA     3         2       3
#> 3     5     2     3     4     1         3       4
#> 4     5     3     1     2     4         4       2
#> 5     3     1     2     4     5         5       4

# use tidyselect helpers
my_coalesce_func(dat = my_df, starts_with("col"), -(col_a:col_c))
#>   col_a col_b col_c col_d col_e extra_col new_col
#> 1     2    NA     4     1     3         1       1
#> 2     2     4     5    NA     3         2       3
#> 3     5     2     3     4     1         3       4
#> 4     5     3     1     2     4         4       2
#> 5     3     1     2     4     5         5       4

由reprex 包于 2021-07-01 创建(v1.0.0)

于 2021-07-01T15:42:28.880 回答