R语言如何用case_when返回所有满足的互容条件组成的列表列？

case_when的设计为短路求值逻辑，匹配到第一个成立的条件就会终止判断，因此无法直接返回多个满足条件的结果。你可以通过「先批量判断所有条件的成立状态，再按行收集符合条件的case名称」的逻辑实现需求，无需手动枚举所有相容条件组合。

实现方案 #

方案1：基于purrr批量判断（推荐，新增条件无需修改主逻辑） #

先把所有条件集中定义在列表中，后续新增条件只需要在列表中追加即可：

library(dplyr)
library(purrr)

# 集中定义所有条件，key为返回的case名称，value为判断逻辑
conditions <- list(
  case1 = ~ .x$mpg >= 15 & .x$cyl == 6,
  case2 = ~ .x$mpg >= 10,
  case3 = ~ .x$cyl >= 7,
  case4 = ~ .x$mpg >= 10,
  case5 = ~ .x$mpg >= 10 & .x$cyl >= 5
)

res <- mtcars %>%
  mutate(
    x = pmap(., ~ {
      cur_row <- list(...)
      # 保留所有判断为TRUE的条件，提取对应的case名称
      names(keep(conditions, \(f) f(cur_row)))
    })
  )

输出结果中的x为列表列，每个元素是当前行满足的所有case名称组成的字符向量。

方案2：基于rowwise逐行处理（无需额外掌握purrr语法） #

先为每个条件生成对应的逻辑判断列，再逐行提取符合条件的列名：

library(dplyr)

res <- mtcars %>%
  # 生成所有条件的逻辑列
  mutate(
    case1 = mpg >= 15 & cyl == 6,
    case2 = mpg >= 10,
    case3 = cyl >= 7,
    case4 = mpg >= 10,
    case5 = mpg >= 10 & cyl >= 5
  ) %>%
  rowwise() %>%
  mutate(
    # 提取当前行值为TRUE的列名，包装为列表
    x = list(names(which(c_across(case1:case5))))
  ) %>%
  # 移除中间生成的逻辑列
  select(-c(case1:case5)) %>%
  ungroup()

可选扩展 #

如果不需要列表列，想要直接返回逗号分隔的字符串，只需要把list(names(which(...)))替换为paste(names(which(...)), collapse = ",")即可。

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