R语言中基于多规则生成分组的更优实现方法探寻

问题说明 #

我有一个包含多列的数据集，每行的列值组合决定新列的生成规则。这些规则组合形式多样，部分规则不涉及所有列，还有列包含较长的生物名称。使用case_when实现时代码杂乱，规则审阅和维护十分繁琐，需要更简洁易维护的实现方式。

测试数据集 #

# 虚拟测试数据集
test <- data.frame(
  col1 = c(1,2,3,4,5),
  col2 = c("A","B","C","D","E"),
  col3 = c(43,22,15,100,60),
  col4 = c("string1","string2","string3","string4","string5"),
  col5 = c("AA","BB","CC","DD","EE"),
  col6 = c("verylongnamehere", "anotherlongname","yetanotherlongname","hereisanotherlongname","thisisthelastlongname")
)

当前实现（case_when） #

library(dplyr)

test2 <- test %>%
  mutate(new_col = case_when(
    col1 == 1 & col2 == "A" & col6 == "verylongnamehere" ~ "result1",
    col3 >= 60 & col5 == "DD" ~ "result2",
    col1 %in% c(2,3,4) & 
     col2 %in% c("B","D") & 
     col5 %in% c("BB","CC","DD") & 
     col6 %in% c("anotherlongname","yetanotherlongname") ~ "result3",
    TRUE ~ "result4"
  ))

优化方案：规则配置表分离 #

核心思路是将规则从业务代码中抽离成独立的配置表，实现规则与逻辑分离，大幅提升可读性和可维护性。

方案1：字符串规则表（直观易上手） #

步骤1：定义规则表

将所有规则集中存储，指定优先级（数字越小越先匹配）：

library(dplyr)
library(purrr)

# 规则配置表
rules <- tibble(
  priority = c(1, 2, 3),
  condition = c(
    "col1 == 1 & col2 == 'A' & col6 == 'verylongnamehere'",
    "col3 >= 60 & col5 == 'DD'",
    "col1 %in% c(2,3,4) & col2 %in% c('B','D') & col5 %in% c('BB','CC','DD') & col6 %in% c('anotherlongname','yetanotherlongname')"
  ),
  result = c("result1", "result2", "result3")
)

# 默认结果
default_result <- "result4"

步骤2：匹配逻辑实现

逐行检查数据是否满足规则，取第一个匹配的结果：

test_optimized <- test %>%
  rowwise() %>%
  mutate(
    # 找到第一个满足的规则条件
    matched_condition = detect(rules$condition, ~ eval(parse(text = .x))),
    # 映射对应结果，无匹配则用默认值
    new_col = ifelse(is.null(matched_condition), default_result, rules$result[rules$condition == matched_condition])
  ) %>%
  ungroup() %>%
  select(-matched_condition)  # 移除中间辅助列

方案2：表达式规则表（更安全） #

如果担心eval(parse)的安全风险，可使用rlang包构造表达式规则，避免字符串解析：

library(dplyr)
library(purrr)
library(rlang)

# 安全规则表：用表达式替代字符串
rules_safe <- tibble(
  priority = c(1, 2, 3),
  condition = list(
    expr(col1 == 1 & col2 == "A" & col6 == "verylongnamehere"),
    expr(col3 >= 60 & col5 == "DD"),
    expr(col1 %in% c(2,3,4) & col2 %in% c("B","D") & col5 %in% c("BB","CC","DD") & col6 %in% c("anotherlongname","yetanotherlongname"))
  ),
  result = c("result1", "result2", "result3")
)

# 匹配逻辑
test_safe <- test %>%
  rowwise() %>%
  mutate(
    # 找到第一个满足的规则索引
    match_idx = which(map_lgl(rules_safe$condition, ~ eval_tidy(.x, data = cur_data())))[1],
    # 映射结果
    new_col = ifelse(is.na(match_idx), default_result, rules_safe$result[match_idx])
  ) %>%
  ungroup() %>%
  select(-match_idx)

方案优势 #

• 规则集中管理：所有规则清晰展示，修改或新增规则只需调整配置表，无需改动主逻辑
• 可读性提升：避免case_when的冗长嵌套，规则审阅更高效
• 扩展性强：新增规则只需在配置表中添加一行，无需重构代码

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