Início / coding / Perguntas / 79553270
Accepted
Nevil
Asked: 2025-04-03 22:45:01 +0800 CST

Achatamento de listas dentro de um dataframe dentro de um dataframe, preservando nomes

  • 772

Apresento aqui um inputdata frame que contém listas de data frames que contêm listas. Algumas das listas de nível inferior estão vazias e algumas listas têm comprimento maior que um. Estou procurando algum código R que transformará o inputem output(também fornecido abaixo)

input = structure(list(pet = c("colin", "fred", "roy"),
                       fruit = list(structure(list(apple = "red",
                                                   banana = "yellow", 
                                                   mango = "green"),
                                              class = "data.frame",
                                              row.names = 1L), 
                                    structure(list(apple = "mouldy",
                                                   banana = "bruised",
                                                   mango = "cut"),
                                              class = "data.frame",
                                              row.names = 1L), 
                                    structure(list(apple = c("windfall", "cooking"),
                                                   banana = c("picked", "ripe"),
                                                   mango = c("stolen", "round")),
                                              class = "data.frame",
                                              row.names = 1:2)), 
                       flavours = list(structure(list()),
                                       structure(list(sweet = "very",
                                                      sour = "ouch", 
                                                      spicy = "hot"),
                                                 class = "data.frame",
                                                 row.names = 1L), 
                                       structure(list(sweet = c("sugary", "calories"),
                                                      sour = c("citrus", "lemon"),
                                                      spicy = c("inferno", "burning")),
                                                 class = "data.frame",
                                                 row.names = 1:2))),
                  row.names = c(NA, 3L),
                  class = "data.frame")

output = data.frame(pet = c("colin", "fred", "roy", "roy"),
                    fruit.apple = c("red", "mouldy", "windfall", "cooking"),
                    fruit.banana = c("yellow", "bruised", "picked", "ripe"),
                    fruit.mango = c("green", "cut", "stolen", "round"),
                    flavours.sweet = c(NA, "very", "sugary", "calories"),
                    flavours.sour = c(NA, "ouch", "citrus", "lemon"),
                    flavours.spicy = c(NA, "hot", "inferno", "burning"))

Os recursos devem ser que o outputdata frame tenha nomes de colunas que sejam a concatenação dos nomes já existentes no input, separados por um ponto. Onde o data frame interno tem uma lista vazia, isso deve dar origem a NA no data frame final e não gerar erros. Onde qualquer um dos data frames internos tem uma lista com comprimento maior que um, esse comprimento será correspondido em todas as listas naquela linha (é um design dos dados de entrada) e deve dar origem a duas linhas no outputcom os valores correspondentes.

Já tentei uma ampla combinação de abordagens usando coisas como rrapply::rrapply(), data.table::'s .SDcol dentro de lapply, unlist() e todas as funções flatten() que pude encontrar, de purrr:: a jsonlite:: ! Nenhuma funcionou para mim até agora.

Encontrei alguns links do Stack Overflow ( aqui e aqui ) que chegaram perto do que eu queria, mas nenhum entregou os nomes de colunas corretos, lidando com listas vazias e listas maiores que um de uma só vez.

Você pode ajudar, por favor? Obrigado.

2 respostas

  1. Best Answer

    Aqui está uma maneira de fazer isso comtidyr::unnest

    library(tidyr)

input |>
  unnest_wider(col = c(fruit, flavours),
               names_sep = '.') |>
  unnest(cols = -pet)
#> # A tibble: 4 × 7
#>   pet   fruit.apple fruit.banana fruit.mango flavours.sweet flavours.sour
#>   <chr> <chr>       <chr>        <chr>       <chr>          <chr>        
#> 1 colin red         yellow       green       <NA>           <NA>         
#> 2 fred  mouldy      bruised      cut         very           ouch         
#> 3 roy   windfall    picked       stolen      sugary         citrus       
#> 4 roy   cooking     ripe         round       calories       lemon        
#> # ℹ 1 more variable: flavours.spicy <chr>
    • 1

  2. Você precisa replace list()com quantidade suficiente de NA.

    > cbind(
+   pet=rep(input$pet, sapply(input$fruit, nrow)),
+   fruit=do.call(rbind, input$fruit),
+   flavours=do.call(rbind, replace(input$flavours, lengths(input$flavours) == 0, list(rep_len(NA, nrow(input)))))
+ )
    pet fruit.apple fruit.banana fruit.mango flavours.sweet flavours.sour flavours.spicy
1 colin         red       yellow       green           <NA>          <NA>           <NA>
2  fred      mouldy      bruised         cut           very          ouch            hot
3   roy    windfall       picked      stolen         sugary        citrus        inferno
4   roy     cooking         ripe       round       calories         lemon        burning

    Você pode encapsular isso em uma função:

    > fx <- \(x) {do.call(rbind, replace(x, lengths(x) == 0, list(rep_len(NA, length(x)))))}
> cbind(pet=rep(input$pet, sapply(input$fruit, nrow)), do.call(cbind, lapply(input[-1], fx)))
    pet fruit.apple fruit.banana fruit.mango flavours.sweet flavours.sour flavours.spicy
1 colin         red       yellow       green           <NA>          <NA>           <NA>
2  fred      mouldy      bruised         cut           very          ouch            hot
3   roy    windfall       picked      stolen         sugary        citrus        inferno
4   roy     cooking         ripe       round       calories         lemon        burning
    • 1

relate perguntas