Início / coding / Perguntas / 79101313
Accepted
deschen
Asked: 2024-10-18 16:52:05 +0800 CST

extrair dinamicamente elementos da coluna da lista

  • 772

Tenho os seguintes dados:

df <- structure(list(id = c("1358792", "1358792", "333482", "333482", "747475", "747475"),
                     x = c("123", "123", "456", "456", NA, NA),
                     all_x = list("123", "123",
                                  c("456", "789"),
                                  c("456", "789"),
                                  list(),
                                  list())),
                row.names = c(NA, -6L),
                class = "data.frame")
    
       id    x    all_x
1 1358792  123      123
2 1358792  123      123
3  333482  456 456, 789
4  333482  456 456, 789
5  747475 <NA>     NULL
6  747475 <NA>     NULL

A coluna all_x é uma lista com um valor VAZIO/NULO, um único caractere ou um vetor de caracteres.

Quero criar uma nova coluna (estilo tidyverse) com a seguinte lógica: quando a all_xcoluna tiver um ou nenhum valor, basta pegar o valor de x. Se tiver dois valores (ou seja, for um vetor de caracteres), queremos agrupar por ide pegar o elemento que corresponde ao número da linha, ou seja, para o primeiro valor id, pegar o primeiro elemento do vetor de caracteres, para o segundo elemento id, pegar o segundo valor de caractere e assim por diante.

A saída desejada seria uma coluna de caracteres adicional com os respectivos valores, ou seja

       id    x    all_x   x2
1 1358792  123      123  123
2 1358792  123      123  123
3  333482  456 456, 789  456
4  333482  456 456, 789  789
5  747475 <NA>     NULL <NA>
6  747475 <NA>     NULL <NA>

Eu tentei várias variantes com if_else, ifelsee deslistagem e indexação, mas ainda assim sempre recebo erros devido à estrutura mista da all_xcoluna.

Aqui está o mais próximo que cheguei:

library(tidyverse)
df |>
  mutate(x2 = if_else(lengths(all_x) > 1, all_x[[1]][row_number()], x), .by = id)

No entanto, obviamente, não tenho sucesso.

3 respostas

  1. Aqui está uma função que retornará xquando all_xtiver um valor ou nenhum, ou então pegará o elemento all_xque corresponde ao número da linha:

    get_x2 <- function(x, all_x, row_num) {
    if (length(all_x) <= 1) {
        return(x)
    }
    all_x[[row_num]]
}

    Então é só criar o row_number()by ide usar Map():

    df |>
    mutate(row_num = row_number(), .by = id) |>
    mutate(
        x2 = Map(get_x2, x, all_x, row_num)
    )
#        id    x    all_x row_num  x2
# 1 1358792  123      123       1 123
# 2 1358792  123      123       2 123
# 3  333482  456 456, 789       1 456
# 4  333482  456 456, 789       2 789
# 5  747475 <NA>     NULL       1  NA
# 6  747475 <NA>     NULL       2  NA
    • 2
  2. Best Answer

    Acho que você deve usar ifelse(em vez de if_else), com o mínimo de esforço para fazê-lo voar

    > df |>
+     mutate(x2 = ifelse(lengths(all_x) > 1, all_x[[1]][row_number()], x), .by = id)
       id    x    all_x   x2
1 1358792  123      123  123
2 1358792  123      123  123
3  333482  456 456, 789  456
4  333482  456 456, 789  789
5  747475 <NA>     NULL <NA>
6  747475 <NA>     NULL <NA>

    Nota: a diferença entre if_elseeifelse

    • 2

  3. Trabalhar com NULLé sempre um pouco desajeitado e geralmente só funciona em um contexto de lista. Você pode substituir NULLpor NA, dessa forma ninguém pode reclamar NULLem um contexto que não seja de lista

    df %>% 
  mutate(x2 = replace(all_x, lengths(all_x) == 0, NA), 
         x2 = unlist(x2)[row_number()], .by = id)
       id    x    all_x   x2
1 1358792  123      123  123
2 1358792  123      123  123
3  333482  456 456, 789  456
4  333482  456 456, 789  789
5  747475 <NA>     NULL <NA>
6  747475 <NA>     NULL <NA>
    • 2

relate perguntas