主页 / coding / 问题 / 77861191
Accepted
Cloft X
Asked: 2024-01-23 00:05:27 +0800 CST

如何为R中组的最后一行赋值？

  • 772

与这个问题类似，我想选择每个组的最后一行并为其分配一个值。

 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

我想要的输出：

 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

我该怎么做呢？任何帮助表示赞赏。

4 个回答

  1. Best Answer

    使用group_by()from dplyr，您可以识别行号与该组的行数相同的情况，并为该观察给出 1，否则为 0。

    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

    创建于 2024-01-22，使用reprex v2.0.2

    • 2

  2. 确定组length并存储在 中len，创建与对应的长度和元素数量integer的空值向量。由ID * 在ID上完成。lenreplacelen1Lave

    > 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

    数据：

    > 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. 这是使用的另一个选项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. 基本 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)

    和 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

相关问题