如何創建按組匯總行的列-有解無憂

資料示例如下：

A和B上面是原始資料，C是我要創建的列。C是基于相同且相鄰的A值的B的組和。即使有一些A = 0，如果它們不相鄰，則不應相加。

uj5u.com熱心網友回復：

tidyverse

library(tidyverse)
df <- data.frame(
           A = c(0L, 0L, 1L, 1L, 0L, 0L, 1L, 1L),
           B = c(1L, 1L, 10L, 5L, 2L, 3L, 20L, 30L)
       )

df %>% 
  group_by(grp = data.table::rleid(A)) %>% 
  mutate(res = sum(B)) %>% 
  ungroup() %>% 
  select(-grp)
#> # A tibble: 8 × 3
#>       A     B   res
#>   <int> <int> <int>
#> 1     0     1     2
#> 2     0     1     2
#> 3     1    10    15
#> 4     1     5    15
#> 5     0     2     5
#> 6     0     3     5
#> 7     1    20    50
#> 8     1    30    50

^{由reprex 包于 2022-04-27 創建(v2.0.1)}

資料表

library(data.table)

setDT(df)[, res := sum(B), by = rleid(A)][]
#>    A  B res
#> 1: 0  1   2
#> 2: 0  1   2
#> 3: 1 10  15
#> 4: 1  5  15
#> 5: 0  2   5
#> 6: 0  3   5
#> 7: 1 20  50
#> 8: 1 30  50

^{由reprex 包于 2022-04-27 創建(v2.0.1)}

根據

df$res <- with(df, ave(B, data.table::rleid(A), FUN = sum))

df
#>   A  B res
#> 1 0  1   2
#> 2 0  1   2
#> 3 1 10  15
#> 4 1  5  15
#> 5 0  2   5
#> 6 0  3   5
#> 7 1 20  50
#> 8 1 30  50

^{由reprex 包于 2022-04-27 創建(v2.0.1)}

uj5u.com熱心網友回復：

library(tidyverse)

data <- tribble(
  ~A, ~B, ~C,
  0, 1, 2,
  0, 1, 2,
  1, 10, 15,
  1, 5, 15,
  0, 2, 5,
  0, 3, 5,
  1, 20, 50,
  1, 30, 50
)
data <- data %>% select(-C)

rle_group <- function(x) {
  rle <- rle(x)
  rle$values <- rle$values %>%
    length() %>%
    seq()
  inverse.rle(rle)
}

data %>%
  mutate(
    group = rle_group(A)
  ) %>%
  group_by(group) %>%
  mutate(
    C = sum(B)
  )
#> # A tibble: 8 × 4
#> # Groups:   group [4]
#>       A     B group     C
#>   <dbl> <dbl> <int> <dbl>
#> 1     0     1     1     2
#> 2     0     1     1     2
#> 3     1    10     2    15
#> 4     1     5     2    15
#> 5     0     2     3     5
#> 6     0     3     3     5
#> 7     1    20     4    50
#> 8     1    30     4    50

^{由reprex 包（v2.0.0）于 2022-04-27 創建}

uj5u.com熱心網友回復：

dplyr沒有任何rle功能的解決方案：

library(dplyr)

df %>%
  group_by(grp = cumsum(c(TRUE, diff(A) != 0))) %>%
  mutate(C = sum(B)) %>%
  ungroup() %>%
  select(-grp)

# A tibble: 8 x 3
      A     B     C
  <int> <int> <int>
1     0     1     2
2     0     1     2
3     1    10    15
4     1     5    15
5     0     2     5
6     0     3     5
7     1    20    50
8     1    30    50

它的base等價物：

within(df, C <- ave(B, cumsum(c(TRUE, diff(A) != 0)), FUN = sum))

  A  B  C
1 0  1  2
2 0  1  2
3 1 10 15
4 1  5 15
5 0  2  5
6 0  3  5
7 1 20 50
8 1 30 50

轉載請註明出處，本文鏈接：https://www.uj5u.com/gongcheng/465575.html

標籤：r 数据框

上一篇：如何有條件地洗掉熊貓資料框的前N??行

下一篇：根據范圍將日期時間分類到新列中