如何将数据框转换为多层嵌套数据框适配eCharts4r树状可视化

需求说明 #

需要将基于兄弟会/姐妹会家谱概念的最高法院大法官数据框，重构为多层世代嵌套结构，适配eCharts4r的树状可视化要求：

• 包含创始人及所有层级的“littles”子数据框
• 无“little”的成员，对应子节点位置为NA，而非无效的子结构

示例数据 #

tree_data <- data.frame(
      name = c("John Jay", "John Rutledge", "William Cushing", "James Wilson", "John Blair", "James Iredell", "Thomas Johnson", "William Paterson", "Samuel Chase", "Oliver Ellsworth", "Bushrod Washington", "Alfred Moore", "John Marshall", "William Johnson", "John McLean", "Levi Woodbery", "William Strong"),
      big = c(NA, "John Jay", "John Jay", "William Cushing", "William Cushing", "John Rutledge", "John Rutledge", "John Rutledge", "John Jay", "Samuel Chase", "James Iredell", "William Cushing", "John Blair", "Bushrod Washington", "Bushrod Washington", "John McLean", "William Johnson"),
      pledgeClass = c("Founder","A", "A", "B", "B", "B", "B", "B", "B", "C", "C", "C", "C", "D", "D", "E", "E"),
      alumniStatus = c(TRUE, TRUE, TRUE, TRUE, FALSE, TRUE, FALSE, TRUE, TRUE, TRUE, FALSE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE))

现有代码问题 #

之前的递归函数在成员无“little”时，会生成包含单个NA的tibble，导致children列类型异常，且可视化时会显示这些无效节点：

create_nested_structure <- function(data, big_boy) {
  data %>% 
    filter(big %in% big_boy) -> d
  if (nrow(d) == 0) {
    return(as.data.frame(NA))
  } else {
    return(mutate(d, children = map(name, ~create_nested_structure(data, .x))))
  }
}

修正后的解决方案 #

调整递归终止逻辑，无节点时返回NULL并替换为NA，确保结构符合要求：

library(dplyr)
library(purrr)

create_nested_structure <- function(data, big_boy) {
  # 筛选当前big对应的所有little成员
  d <- data %>% filter(big %in% big_boy)
  
  if (nrow(d) == 0) {
    # 无后续节点时返回NULL，避免生成无效数据框
    return(NULL)
  } else {
    # 递归生成子节点，并将NULL转换为NA以保证列类型一致
    d %>% 
      mutate(children = map(name, ~create_nested_structure(data, .x)) %>% 
               map_if(is.null, ~NA))
  }
}

# 从创始人（big字段为NA的节点）开始构建嵌套结构
nested_tree <- create_nested_structure(tree_data, NA)

修改要点 #

1. 终止条件优化：无节点时返回NULL，而非含NA的数据框，避免生成无效子结构
2. 类型一致性处理：使用map_if将递归返回的NULL替换为NA，确保children列类型统一，不会出现布尔型异常
3. 根节点调用：以NA作为初始参数，对应创始人的big字段值，正确生成从根节点开始的完整树结构

验证结果 #

生成的nested_tree中，无“little”的成员（如James Wilson）的children列值为NA，而非含NA的tibble，完全适配eCharts4r树状可视化的输入要求。

内容的提问来源于stack exchange，提问作者Logan W.