1

我有一个数据框。片段是：

df1 <- data.frame(x = c(1, 2, 1, 3, 5, 1, 4, 1), y = c(1, 1, 2, 2, 1, 1, 1, 3))
  x y
1 1 1
2 2 1
3 1 2
4 3 2
5 5 1
6 1 1
7 4 1
8 1 3

我需要对 df1 进行分组y并求和，x但要考虑 y 的顺序。即我需要在每个新组之后创建新组，y并对相应x的 . 所需的输出是

   x y
1  3 1
2  4 2
3 10 1
4  1 3

如何在 R 中做到这一点？

4

2 回答 2

2

我们可以使用rleid(from data.table) 来获取 run-length-id 以对相邻的相似元素进行分组，并获取sum'x'

library(dplyr)
library(data.table)
df1 %>%
   group_by(grp= rleid(y), y) %>%
   summarise(x = sum(x)) %>%
   ungroup %>%
   select(names(df1))
# A tibble: 4 x 2
#      x     y
#  <dbl> <dbl>
#1     3     1
#2     4     2
#3    10     1
#4     1     3

或仅使用 'y' 值dplyr创建一个逻辑表达式，使用inlag转换为数字索引并获取'x'cumsumgroup_bysum

df1 %>%
   group_by(grp = cumsum(y != lag(y, default = first(y)))) %>% 
   summarise(x = sum(x), y = first(y)) %>%
   select(-grp)
于 2019-11-07T19:41:29.143 回答
1

使用data.table：

library(data.table)

setDT(df1)[, .(x=sum(x), y=y[1]), .(rleid(y))][,rleid:=NULL][]

#>     x y
#> 1:  3 1
#> 2:  4 2
#> 3: 10 1
#> 4:  1 3

或使用其他dplyr解决方案base::rle：

library(dplyr)

df1 %>% 
  group_by(y_grp = with(rle(y), rep(seq_along(lengths), lengths))) %>% 
  summarise(x = sum(x), y = y[1]) %>% 
  ungroup %>% select(-y_grp)

#> # A tibble: 4 x 2
#>       x     y
#>   <dbl> <dbl>
#> 1     3     1
#> 2     4     2
#> 3    10     1
#> 4     1     3
于 2019-11-07T19:56:10.103 回答