如何在R的data.table中展开嵌套列表并匹配对应位置

首先构造你提供的原数据：

library(data.table)
dt <- data.table(
  value = "one",
  list1 = list(list("one1", "one2", c("one3 1", "one3 2"))),
  position = list(c(0, 1, 2))
)

正确展开方法 #

核心思路是先让list1的每个子元素与position的对应值配对，再展开子元素内部的向量，同时重复对应的position值：

方法一：分步data.table操作

# 第一步：拆分list1为子元素，与position的每个值一一对应
dt_step1 <- dt[, .(sub_list = unlist(list1, recursive = FALSE), pos = position[[1]]), by = value]

# 第二步：展开每个子元素，保留对应position
result <- dt_step1[, .(list1 = unlist(sub_list)), by = .(value, pos)]
setnames(result, "pos", "position")

方法二：一步式嵌套操作

result <- dt[, {
  # 将list1子元素与position值配对
  element_pos_pairs <- Map(list, sub_list = unlist(list1, recursive = FALSE), pos = position[[1]])
  # 展开每个配对组
  rbindlist(lapply(element_pos_pairs, function(x) data.table(list1 = x$sub_list, position = x$pos)))
}, by = value]

结果验证 #

执行后得到的result就是你需要的格式：

value     list1 position
1:   one      one1        0
2:   one      one2        1
3:   one one3 1          2
4:   one one3 2          2

错误原因说明 #

你之前尝试出现one3 2对应position=3，是因为直接同时展开list1和position时，list1展开后长度为4，而position原长度为3，data.table会自动循环填充position值，导致最后一个元素匹配到循环后的3。通过先配对子元素与position，再展开子元素，就能避免这种错误匹配。

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