如何在R中基于另一数据框提取分组内指定行的下一行？

假设你使用dplyr工具包，完全可以通过提取目标行下一行的ID_IA向量，结合filter函数实现需求。以下是具体实现方案，先构造示例数据方便理解：

library(dplyr)

# 示例数据框
df1 <- tibble(
  ID2 = rep(c("A", "B"), each = 4),
  ID_IA = c("A1", "A2", "A3", "A4", "B1", "B2", "B3", "B4"),
  ia_label = c("word1", "word2", "word3", "word4", "word5", "word6", "word7", "word8")
)

df2 <- tibble(
  ID2 = c("A", "B"),
  ID_IA = c("A2", "B3")
)

方法一：用lead生成下一行ID，结合filter筛选 #

1. 先在df1的每个ID2分组内，用lead函数获取每行的下一行ID_IA：

df1_with_next <- df1 %>%
  group_by(ID2) %>%
  mutate(next_ID_IA = lead(ID_IA)) %>% # 生成当前行的下一行ID_IA
  ungroup()

2. 将df2与处理后的df1关联，提取所有目标行对应的下一行ID_IA（转为向量），再用filter筛选df1中的对应行：

# 提取目标下一行的ID_IA向量
target_ids <- df1_with_next %>%
  inner_join(df2, by = c("ID2", "ID_IA")) %>%
  filter(!is.na(next_ID_IA)) %>% # 排除目标行是分组最后一行的情况
  pull(next_ID_IA)

# 筛选得到df3
df3 <- df1 %>%
  filter(ID_IA %in% target_ids)

运行后df3的结果就是每个ID2分组中df2指定行的下一行数据：

# A tibble: 2 × 3
  ID2  ID_IA ia_label
  <chr> <chr> <chr>
1 A     A3    word3
2 B     B4    word8

方法二：用行号定位目标行 #

如果需要更明确地通过行顺序定位，也可以给每个分组添加行号，再找到目标行的下一行：

# 给df1每个分组添加行号
df1_with_row <- df1 %>%
  group_by(ID2) %>%
  mutate(row_num = row_number()) %>%
  ungroup()

# 找到df2对应行的行号，计算目标行号（行号+1）
target_rows <- df1_with_row %>%
  inner_join(df2, by = c("ID2", "ID_IA")) %>%
  mutate(target_row = row_num + 1) %>%
  select(ID2, target_row)

# 匹配目标行号得到df3
df3 <- df1_with_row %>%
  inner_join(target_rows, by = c("ID2", "row_num" = "target_row")) %>%
  select(-row_num)

注意事项 #

• 确保df1的每个ID2分组内的行顺序是你需要的逻辑顺序（比如按ID_IA排序），如果顺序不对，需要先在分组内用arrange排序后再处理。
• 如果df2中某ID2的指定行是该分组的最后一行，lead会返回NA，此时需要根据需求决定是否保留或过滤这类情况。

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