Question

我有一组用于使用dplyr生成汇总统计信息的命令。

我想知道正在使用哪些数据列。

数据列的前缀为.data [[＆＃34; ColumnName＆＃34;]]。

例如我们有：

my_quos <- rlang::list2(
  "GenderD" = rlang::quo(length(.data[["TeamCode"]])),
  "GenderMaleN" = rlang::quo(.data[["S1IsMale"]])
)

我已经开始使用rlang :: call_args（）将命令分解为其组件来解决此问题：

my_args_test <- rlang::call_args(my_quos[[1]])
str(my_args_test)
List of 1
 $ : language .data[["TeamCode"]]

列应该全部作为数据代词。有没有快速的方法来检查列表中的项目是否是数据代词？我试过了：

is(my_args_test[[1]], "rlang_data_pronoun")

但这会返回false。将字符串检查为以.data开头的文本[[我猜可能是一个选项（但我怀疑它更容易出错）。

还有办法干净地返回传递给数据代词的参数而不是解析字符串吗？换句话说，目标是理想地将我们的输出返回为：

c("TeamCode", "S1IsMale")

从原来的my_quos。

Answer 1

这可以分两个步骤完成。首先，您要提取您的定量捕获的表达式并将其转换为Abstract Syntax Trees (ASTs)。

## Recursively constructs Abstract Syntax Tree for a given expression
getAST <- function( ee ) { as.list(ee) %>% purrr::map_if(is.call, getAST) }

## Apply function to expressions captured by each quosure
asts <- purrr::map( my_quos, quo_get_expr ) %>% purrr::map( getAST )
str(asts)
# List of 2
#  $ GenderD    :List of 2
#   ..$ : symbol length
#   ..$ :List of 3
#   .. ..$ : symbol [[
#   .. ..$ : symbol .data
#   .. ..$ : chr "TeamCode"
#  $ GenderMaleN:List of 3
#   ..$ : symbol [[
#   ..$ : symbol .data
#   ..$ : chr "S1IsMale"

从这里，我们看到匹配.data[["somename"]]的模式是一个长度为3的列表，其中第一个条目为[[，第二个条目为.data，最后一个条目是您所需要的正在尝试提取。让我们编写一个识别此模式并在识别后返回第三个元素的函数（注意：此函数显示如何将项与.data代词匹配，这是您的另一个问题）：

## If the input matches .data[["name"]], returns "name". Otherwise, NULL
getName <- function( x )
{
  if( is.list(x) && length(x) == 3 &&          ## It's a length-3 list
      identical( x[[1]], quote(`[[`) ) &&      ##  with [[ as the first element
      identical( x[[2]], quote(.data) ) &&     ##  .data as the second element
      is.character(x[[3]]) ) return(x[[3]])    ##  and a character string as 3rd
  NULL
}

使用此功能，第二步只是简单地将其递归应用于AST列表，以提取使用的列名。

getNames <- function( aa ) { 
  purrr::keep(aa, is.list) %>% 
  purrr::map(getNames) %>%            ## Recurse to any list descendants
  c( getName(aa) ) %>%                ## Append self to the result
  unlist                              ## Return as character vector, not list
}

getNames(asts)
#     GenderD GenderMaleN 
#  "TeamCode"  "S1IsMale"

使用Rlang：在一组排序

1 个答案: