基于dplyr的变量乘选及可选分组自定义函数开发求助

完全没问题！这正是tidy evaluation（tidyeval）的拿手场景——帮你在dplyr里处理动态变量，同时避免重复写相似的代码逻辑。我会给你实现一个符合需求的自定义函数，同时拆解背后的关键逻辑。

首先先确认我们的测试数据集：

library(ggplot2)
library(dplyr) # 别忘了加载dplyr核心包！
msleep_mini <- msleep[1:10, ]

接下来是自定义函数的实现，我把它命名为transform_dbl_var，逻辑完全贴合你的需求：

transform_dbl_var <- function(.data, var, group_var = NULL) {
  # 捕获目标变量和分组变量（如果存在）
  var_enquo <- enquo(var)
  group_enquo <- enquo(group_var)
  
  # 定义乘2后的变量名（比如sleep_total变成sleep_total_doubled）
  new_var_name <- paste0(quo_name(var_enquo), "_doubled")
  
  # 基础操作：保留指定列 + 目标变量乘2
  result <- .data %>%
    select(name, vore, order, !!var_enquo) %>%
    mutate(!!new_var_name := !!var_enquo * 2) %>%
    select(-!!var_enquo) # 移除原始变量，只保留乘2后的列
  
  # 判断是否有分组参数
  if (!quo_is_null(group_enquo)) {
    result <- result %>%
      group_by(!!group_enquo) %>%
      arrange(!!sym(new_var_name)) %>% # 按乘2后的变量排序
      mutate(id = row_number()) %>%
      ungroup()
  }
  
  return(result)
}

关键逻辑解释 #

• enquo() 和 !!：enquo()用来捕获用户传入的变量名（把它转换成tidyeval可识别的表达式），!!（俗称"bang-bang"）用来把捕获的表达式注入到dplyr函数中，让dplyr能正确识别动态变量。
• quo_name() 和 sym()：前者把捕获的变量转换成字符串，用来生成新的列名；后者把字符串转回变量符号，方便在arrange()这类需要变量名的函数中使用。
• 条件分支：通过quo_is_null()判断是否传入了分组变量，从而执行不同的逻辑分支，兼容单/双参数两种调用场景。

测试用例 #

1. 单参数调用（无分组）：

transform_dbl_var(msleep_mini, sleep_total)

返回的结果会包含name、vore、order和sleep_total_doubled列，完全符合你的第一个需求。

2. 双参数调用（带分组）：

transform_dbl_var(msleep_mini, sleep_total, vore)

这时候会按vore分组，对sleep_total_doubled排序，然后添加组内行号id列，满足第二个需求。

3. 支持任意合法变量：
   比如我们换个变量试试，用bodywt作为目标变量，order作为分组变量：

transform_dbl_var(msleep_mini, bodywt, order)

函数一样能正常工作，生成bodywt_doubled列，按order分组排序并添加id。

这个函数完全适配你提到的所有场景，而且因为用了tidyeval，能灵活处理DataFrame里的任意合法变量~

内容的提问来源于stack exchange，提问作者DeltaIV