如何用data.table按组过滤满足条件的分组（保留整组数据）

问题说明 #

需要保留所有包含Year=2009的ID对应的整组数据（哪怕该组仅一行符合条件）。之前尝试的dplyr代码无法实现需求：

filtered <- dt %>% group_by(ID) %>% filter(Year == 2009)

这段代码只会保留组内Year=2009的单行，而非整组数据。

示例数据 #

data1 <- data.frame(
  "ID" = c(1,1,1,1,1,1,2,2,2,2,2,3,3,3,4,4,4,4),
  "Year" = c(2009,2010,2011,2012,2013,2014,2010,2011,2012,2013,2014,2009,2010,2011,2014,2015,2016,2017),
  "Price" = c(4,5,7,3,5,7,9,6,7,8,5,3,5,7,8,9,3,5)
)

data.table实现方案 #

先加载data.table包并转换数据格式，以下两种方法均可实现需求：

方法1：先筛选目标ID，再提取整组

library(data.table)
dt <- as.data.table(data1)
# 找出所有包含Year=2009的ID
target_ids <- dt[Year == 2009, unique(ID)]
# 提取这些ID的全部行
filtered_dt <- dt[ID %in% target_ids]

方法2：分组后直接筛选整组

利用data.table的分组特性，按ID分组后，判断组内是否存在Year=2009，存在则保留整组所有行：

library(data.table)
dt <- as.data.table(data1)
filtered_dt <- dt[, .SD[any(Year == 2009)], by = ID]

结果说明 #

两种方法都会保留ID为1和3的所有行（这两个ID均包含2009年数据），排除ID为2和4的行，完全符合需求。

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