R语言如何合并列表中第二个元素匹配的相邻向量

问题原因 #

你之前用for循环出现下标越界，是因为for循环会预先固定遍历次数为列表原始长度，但合并相邻元素后列表总长度会缩短，后续遍历到不存在的下标就会报错。推荐用动态判断长度的while循环，或者向量化分组的方法实现需求：

方案1：while循环实现（易理解） #

# 构造测试数据
n_vectors <- list(
  c("Hello Another World", "Key_123", "", "Steven", "Robert"),
  c("Hello World", "Key_ABC", "PART1", "John", "Paul"),
  c("Hello World", "Key_ABC", "PART2", "Kim", "Tim"),
  c("Bye World", "Key_456", "", "Mo", "Jordan")
)

i <- 1
# 每次判断都取当前列表的实际长度，不会越界
while (i < length(n_vectors)) {
  cur_key <- n_vectors[[i]][2]
  next_key <- n_vectors[[i+1]][2]
  if (cur_key == next_key) {
    # 合并当前和下一个元素
    n_vectors[[i]] <- c(n_vectors[[i]], n_vectors[[i+1]])
    # 删除已合并的下一个元素
    n_vectors <- n_vectors[-(i+1)]
  } else {
    i <- i + 1
  }
}

# 查看结果
n_vectors

运行结果完全符合你的预期：

[[1]]
[1] "Hello Another World" "Key_123"             ""                    "Steven"              "Robert"             

[[2]]
 [1] "Hello World" "Key_ABC"     "PART1"       "John"        "Paul"        "Hello World" "Key_ABC"     "PART2"       "Kim"         "Tim"        

[[3]]
[1] "Bye World" "Key_456"   ""          "Mo"        "Jordan"

该方案也支持连续多个相邻元素匹配的场景，会自动把所有连续匹配的元素合并为一个。

方案2：向量化分组实现（更简洁高效） #

如果列表长度很大，推荐用分组的向量化操作，不需要手动控制循环下标：

# 提取所有向量的第二个元素作为键
keys <- sapply(n_vectors, `[`, 2)
# 生成连续相同键的分组ID
group_id <- cumsum(c(TRUE, keys[-1] != keys[-length(keys)]))
# 按分组合并向量
result <- tapply(n_vectors, group_id, \(x) do.call(c, x)) |> unname()

运行结果和方案1完全一致。

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