如何在R中对嵌套列表排序并按索引分组提取元素

需求回顾 #

给定嵌套列表，需完成以下操作：

1. 按每个子元素的第三个值排序
2. 重置子元素的第一个索引值，生成Final列表
3. 根据分组索引G，提取Final中对应位置的第二个元素

R代码实现 #

# 原始嵌套列表
L <- list(
  c(0, 'A', 50.1),
  c(1, 'B', 50.0),
  c(2, 'C', 50.2),
  c(3, 'D', 50.7),
  c(4, 'E', 50.3)
)

# 步骤1：按子元素第三个值排序
# 提取每个子元素的第三个数值，生成排序索引
sort_indices <- order(sapply(L, function(x) as.numeric(x[3])))
sorted_L <- L[sort_indices]

# 步骤2：重置索引生成Final列表
# 新索引从0开始，替换每个子列表的第一个元素
Final <- lapply(seq_along(sorted_L), function(i) {
  c(as.character(i-1), sorted_L[[i]][2], sorted_L[[i]][3])
})

# 步骤3：根据分组索引G提取对应元素
G <- list(c(0,1), c(1,3), c(2,3,4))
out <- lapply(G, function(group) {
  # R中列表索引从1开始，需将group中的0-based索引转为1-based
  sapply(group + 1, function(idx) Final[[idx]][2])
})

# 查看结果
print(Final)
# [[1]]
# [1] "0"   "B"   "50.0"
# [[2]]
# [1] "1"   "A"   "50.1"
# [[3]]
# [1] "2"   "C"   "50.2"
# [[4]]
# [1] "3"   "E"   "50.3"
# [[5]]
# [1] "4"   "D"   "50.7"

print(out)
# [[1]]
# [1] "B" "A"
# [[2]]
# [1] "A" "E"
# [[3]]
# [1] "C" "E" "D"

代码说明 #

• 使用sapply提取每个子元素的第三个值，结合order生成排序索引，完成排序操作
• 通过lapply遍历排序后的列表，替换第一个元素为新的0-based索引，生成Final列表
• 处理分组索引时，将R的1-based索引与需求中的0-based索引转换，提取对应位置的第二个元素

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