Início / coding / Perguntas / 79566992
Accepted
Slav
Asked: 2025-04-10 22:52:21 +0800 CST

Quadro de dados em lista aninhada (árvore)

  • 772

Como transformar um quadro de dados em uma lista recursiva de listas (árvore)?

Entrada

roles_data <- data.frame(
  Child = c("A", "B", "C", "D", "E", "F"),
  ParentID = c(NA, "A", "B", "B", "D", "D")  # The top-level role has no parent (NA)
)

Saída desejada

desired_output <- list(
 list(
 text = "A",
 children = list(
 list(
 text = "B",
 children = list(
 list(
 text = "C"),
 list(
 text = "D"
 ,
 children = list(
 list(
  text = "E"
 ), 
  list(
  text = "F"
 )  ) ) ) ) ) )

(Espero não ter mexido na tabela de saída na parte inferior, só preciso ter mais de um nível de profundidade.)

1 respostas

  1. Best Answer

    Você está descrevendo uma estrutura em árvore : cada pai pode ter vários filhos, mas cada filho tem exatamente um pai. Buscando por perguntas sobre árvores, isso é semelhante, em princípio, a " Transformar um dataframe em uma estrutura em árvore" (lista de listas) , no sentido de que podemos começar da raiz e recursivamente sobre os filhos. No entanto, como seus dados de entrada e saída estão em formatos diferentes, podemos fazer algo um pouco mais ergonômico. Basicamente, para cada nó, crie uma lista com text = node_name, e se houver filhos, construa recursivamente uma subárvore para eles.

    generate_tree <- function(dat, root = dat$Child[is.na(dat$ParentID)]) {
    node <- list(text = root)
    children <- na.omit(dat$Child[dat$ParentID == root])
    if (length(children) > 0) {
        node$children <- lapply(children, \(child) generate_tree(dat, child))
    }
    node
}

    Se você realmente quiser a parte externa, list()você pode envolvê-la em uma:

    identical(list(generate_tree(roles_data)), desired_output) 
# [1] TRUE

    E aqui está a saída em json para fins de impressão:

    generate_tree(roles_data)  |> jsonlite::toJSON(pretty = TRUE, auto_unbox = TRUE)
{
  "text": "A",
  "children": [
    {
      "text": "B",
      "children": [
        {
          "text": "C"
        },
        {
          "text": "D",
          "children": [
            {
              "text": "E"
            },
            {
              "text": "F"
            }
          ]
        }
      ]
    }
  ]
}
    • 4

relate perguntas