Início / coding / Perguntas / 77861191
Accepted
Cloft X
Asked: 2024-01-23 00:05:27 +0800 CST

Como atribuir valor à última linha do grupo em R?

  • 772

Semelhante a esta pergunta , quero selecionar a última linha de cada grupo e atribuir um valor a ela.

 a <- data.frame("ID" = c("A", "A", "B", "B", "C", "C"),
                 "NUM" = c(1, 2, 4, 3, 6, 9),
                 "VAL" = c(1, 0, 1, 0, 1, 0))
 
 ID NUM VAL
  A   1   1
  A   2   0
  B   4   1
  B   3   0
  C   6   1
  C   9   0

Minha saída desejada:

 ID NUM VAL end.spell
  A   1   1         0
  A   2   0         1
  B   4   1         0
  B   3   0         1
  C   6   1         0
  C   9   0         1

Como eu faço isso? Qualquer ajuda é apreciada.

4 respostas

  1. Best Answer

    Usando group_by()from dplyr, você pode identificar casos em que o número da linha é igual ao número de linhas desse grupo e atribuir 1 para essa observação e zero caso contrário.

    library(dplyr)
a <- data.frame("ID" = c("A", "A", "B", "B", "C", "C"),
                "NUM" = c(1, 2, 4, 3, 6, 9),
                "VAL" = c(1, 0, 1, 0, 1, 0))


a %>% 
  group_by(ID) %>% 
  mutate(end.spell = ifelse(row_number() == n(), 1,0))
#> # A tibble: 6 × 4
#> # Groups:   ID [3]
#>   ID      NUM   VAL end.spell
#>   <chr> <dbl> <dbl>     <dbl>
#> 1 A         1     1         0
#> 2 A         2     0         1
#> 3 B         4     1         0
#> 4 B         3     0         1
#> 5 C         6     1         0
#> 6 C         9     0         1

    Criado em 22/01/2024 com reprex v2.0.2

    • 2

  2. Determine o grupo lengthe armazene em len, crie integerum vetor de nulos de comprimento lene replacenúmero do elemento correspondente a lenwith 1L. Feito em aveID s por ID s*.

    > transform(a, end=ave(ID, ID, FUN=\(x) {
+   len <- length(x)
+   replace(integer(len), len, 1L)
+ }))
  ID NUM VAL end
1  A   1   1   0
2  A   2   0   1
3  B   4   1   0
4  B   3   0   1
5  C   6   1   0
6  C   9   0   1

    Dados:

    > dput(a)
structure(list(ID = c("A", "A", "B", "B", "C", "C"), NUM = c(1, 
2, 4, 3, 6, 9), VAL = c(1, 0, 1, 0, 1, 0)), class = "data.frame", row.names = c(NA, 
-6L))
    • 0

  3. Aqui está outra opção usando last():

    library(dplyr)
a %>% 
  mutate(end.spell = ifelse(VAL == last(VAL), 1, 0))

      ID NUM VAL end.spell
1  A   1   1         0
2  A   2   0         1
3  B   4   1         0
4  B   3   0         1
5  C   6   1         0
6  C   9   0         1
    • 0

  4. Uma implementação base R:

    a <- data.frame("ID" = c("A", "A", "B", "B", "C", "C"),
            "NUM" = c(1, 2, 4, 3, 6, 9),
            "VAL" = c(1, 0, 1, 0, 1, 0))

max_idxs <- sapply(unique(a$ID), \(x) tail(which(a$ID == x), 1))
a$end.spell <- ifelse(1:nrow(a) %in% max_idxs, 1, 0)

    E uma implementação data.table:

    library(data.table)
a <- data.table("ID" = c("A", "A", "B", "B", "C", "C"),
                "NUM" = c(1, 2, 4, 3, 6, 9),
                "VAL" = c(1, 0, 1, 0, 1, 0))
a[, end.spell := ifelse(1:.N == .N, 1, 0), by = ID]
    • 0

relate perguntas