Question

目标：编写一个将数据框作为第一个的函数参数，然后传递给另外两个参数，即argments列表 dplyr::select这样函数将返回两个数据帧。

这是一个工作示例

my_select <- function(.data, df1, df2) {
  DF1 <- dplyr::select(.data, rlang::UQS(df1))
  DF2 <- dplyr::select(.data, rlang::UQS(df2))
  list(DF1 = DF1, DF2 = DF2)
}

working_eg <-
  my_select(mtcars,
          df1 = alist(dplyr::contains("r"), dplyr::matches("^.p.*")),
          df2 = alist(disp))

str(working_eg, max.length = 1L)

## List of 2
##  $ DF1:'data.frame': 32 obs. of  5 variables:
##   ..$ drat: num [1:32] 3.9 3.9 3.85 3.08 3.15 2.76 3.21 3.69 3.92 3.92 ...
##   ..$ gear: num [1:32] 4 4 4 3 3 3 3 4 4 4 ...
##   ..$ carb: num [1:32] 4 4 1 1 2 1 4 2 2 4 ...
##   ..$ mpg : num [1:32] 21 21 22.8 21.4 18.7 18.1 14.3 24.4 22.8 19.2 ...
##   ..$ hp  : num [1:32] 110 110 93 110 175 105 245 62 95 123 ...
##  $ DF2:'data.frame': 32 obs. of  1 variable:
##   ..$ disp: num [1:32] 160 160 108 258 360 ...

我希望调整df1和df2采用list而不是alist。但是，如果参数为my_select

，函数list将失败

my_select(mtcars,
          df1 = list(dplyr::contains("r"), dplyr::matches("^.p.*")),
          df2 = list(disp))
## Error: Variable context not set

我不想让最终用户使用alist，如果可能的话，因为我没有测试alist用于传递参数而不是list的好方法。

我尝试过几种rlang::UQ，rlang::UQE，rlang::UQS的组合使用rlang::quo，rlang::enquos和rlang::quos来解决此问题。我是什么思想是最好的方法是：

my_select2 <- function(.data, df1, df2) {
  DF1 <- dplyr::select(.data, rlang::UQS(rlang::quos(df1)))
  DF2 <- dplyr::select(.data, rlang::UQS(rlang::quos(df2)))
  list(DF1 = DF1, DF2 = DF2)
}

my_select2(mtcars,
          df1 = list(dplyr::contains("r"), dplyr::matches("^.p.*")),
          df2 = list(disp))
## Error: `df1` must resolve to integer column positions, not a list

有没有办法使用rlang 打包dplyr以便。{ my_select2的语法将返回与my_select时相同的对象参数通过alist传递？

packageVersion("dplyr")
# [1] ‘0.7.4’
packageVersion("rlang")
# [1] ‘0.1.2’

Answer 1

@ MrFlick的评论提醒我需要解决问题并改进api。

my_select3 <- function(.data, df1, df2) {
  DF1 <- dplyr::select(.data, rlang::UQS((df1)))
  DF2 <- dplyr::select(.data, rlang::UQS((df2)))
  list(DF1 = DF1, DF2 = DF2)
}

working_eg2 <-
  my_select3(mtcars,
            df1 = dplyr::vars(dplyr::contains("r"), dplyr::matches("^.p.*")),
            df2 = dplyr::vars(disp))

all.equal(working_eg, working_eg2)
# [1] TRUE

通过rlang quosures和dplyr传递多个参数列表

1 个答案: