R语言：筛选包含指定全部元素的DataFrame ID并返回对应行

原函数问题分析 #

你的函数返回空数据框有两个核心问题：

• 用setequal()对比的两个数据框结构不一致：左边是带ID和values的两列数据，右边只有values单列，永远不会相等；
• setequal()要求两边元素完全匹配，不符合你“至少包含所有指定元素”的需求（比如ID4多了d就会被排除）。

Base R 实现方案 #

以下函数完全满足你的需求，且支持任意长度的items列表扩展：

Criteria.Match <- function(df, CriteriaList, criteria.string) {
  # 第一步：找出所有包含CriteriaList全部元素的ID
  valid_ids <- sapply(unique(df$ID), function(id) {
    # 获取当前ID对应的所有values
    current_values <- df[df$ID == id, criteria.string]
    # 检查CriteriaList的所有元素是否都在当前values中
    all(CriteriaList %in% current_values)
  })
  # 提取符合条件的ID
  valid_ids <- names(valid_ids[valid_ids])
  
  # 第二步：过滤出符合条件的ID，且仅保留values属于CriteriaList的行
  result <- df[df$ID %in% valid_ids & df[[criteria.string]] %in% CriteriaList, ]
  # 重置行名（可选，让结果更整洁）
  rownames(result) <- NULL
  return(result)
}

# 测试示例数据
items <- c("a", "b", "c")
df <- data.frame(
  ID = c(1,2,2,3,3,3,4,4,4,4),
  values = c("b","a","c","a","b","c","a","b","c","d")
)

# 调用函数
Criteria.Match(df, items, 'values')

dplyr 实现方案（更简洁） #

如果平时用tidyverse工具链，这个写法更直观：

library(dplyr)

df_filtered <- df %>%
  # 先过滤掉不在items里的values（可选，提前减少数据量）
  filter(values %in% items) %>%
  # 按ID分组
  group_by(ID) %>%
  # 只保留那些包含所有items元素的分组
  filter(all(items %in% values)) %>%
  # 取消分组
  ungroup()

# 查看结果
df_filtered

两种方案都会返回如下结果：

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