如何在R的tidyverse中创建兼容NULL与非NULL分组参数的计数函数？

你的函数问题出在参数设计和调用逻辑上：原函数用单独的col_name参数接收第一个分组变量，...接收剩余变量，这种设计不仅不符合常规调用习惯，还容易导致逻辑混淆。以下是修正后的实现方案：

修正后的函数 #

library(tidyverse)

# 构造测试数据
tib <- tibble( 
          colB = sample(c("A", "B"), 50, replace = T), 
          colC = sample(c("C", "D"), 50, replace = T),
          colD = sample(c("C", "D"), 50, replace = T)
          )

test_fun <- function(df, ...){
  # 捕获所有传入的分组变量
  groups <- enquos(...)
  
  # 判断是否有分组变量传入
  if(rlang::is_empty(groups)){
    # 无分组变量时返回总行数
    count(df)
  } else {
    # 有分组变量时按分组统计行数
    df %>% count(!!!groups)
  }
}

测试验证 #

# 无分组变量：返回总行数
test_fun(tib)

# 单个分组变量：按colB分组统计
test_fun(tib, colB)

# 多个分组变量：按colB和colC分组统计
test_fun(tib, colB, colC)

原函数问题说明 #

1. 参数设计冗余：单独的col_name参数完全可以被...替代，统一接收所有分组变量更符合R函数的常规设计。
2. 调用逻辑不直观：原函数要求第一个分组变量必须传给col_name，剩余变量传给...，这种写法不符合用户的自然调用习惯，容易出错。
3. 判断逻辑局限：原函数仅通过is.null(col_name)判断是否需要分组，但无法处理用户只传入...而不指定col_name的情况。

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