是否有相当于data.table：rleid的dplyr？

Question

data.table提供了一个很好的方便功能，rleid用于游程编码：library(data.table)DT = data.table(grp=rep(c("A", "B", "C", "A", "B"), c(2, 2, 3, 1, 2)), value=1:10)rleid(DT$grp)# [1] 1 1 2 2 3 3 3 4 5 5我可以在基地模仿这个R有：df <- data.frame(DT)rep(seq_along(rle(df$grp)$values), times = rle(df$grp)$lengths)# [1] 1 1 2 2 3 3 3 4 5 5有没有人知道dplyr等价物(？)或者是创建rleid行为与dplyr是执行如下操作library(dplyr)my_rleid = rep(seq_along(rle(df$grp)$values), times = rle(df$grp)$lengths)df %>%   mutate(rleid = my_rleid)

Answer 1

你可以做(当你两者兼备的时候)数据表和dplyr(已装载)：

DT <- DT %>% mutate(rlid = rleid(grp))

这意味着：

> DT
    grp value rlid 1:   A     1    1
 2:   A     2    1
 3:   B     3    2
 4:   B     4    2
 5:   C     5    3
 6:   C     6    3
 7:   C     7    3
 8:   A     8    4
 9:   B     9    510:   B    10    5

当你不想装载数据表另外，您还可以使用(如@DavidArenburg在注释中提到的)：

DT <- DT %>% mutate(rlid = data.table::rleid(grp))

你可以复制/窃取它：

myrleid <- data.table::rleid

Answer 2

如果你只想用R和dplyr，更好的方法是将您自己的一两行版本的rleid()作为一个函数，然后在需要的时候应用它。

library(dplyr)

myrleid <- function(x) {

    x <- rle(x)$lengths

    rep(seq_along(x), times=x)

}

## Try it out

DT <- DT %>% mutate(rlid = myrleid(grp))

DT

#   grp value rlid

# 1:   A     1    1

# 2:   A     2    1

# 3:   B     3    2

# 4:   B     4    2

# 5:   C     5    3

# 6:   C     6    3

# 7:   C     7    3

# 8:   A     8    4

# 9:   B     9    5

#10:   B    10    5

Answer 3

您可以使用lag功能来自dplyr.

DT <-
    DT %>%
    mutate(rleid = (grp != lag(grp, 1, default = "asdf"))) %>%
    mutate(rleid = cumsum(rleid))

施予

> DT
    grp value rleid 1:   A     1     1
 2:   A     2     1
 3:   B     3     2
 4:   B     4     2
 5:   C     5     3
 6:   C     6     3
 7:   C     7     3
 8:   A     8     4
 9:   B     9     510:   B    10     5