首頁(yè) 猿問(wèn) 是否有相當(dāng)于data.table：...

是否有相當(dāng)于data.table：rleid的dplyr？

R語(yǔ)言

牧羊人nacy 2019-07-13 15:43:58

data.table提供了一個(gè)很好的方便功能，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我可以在基地模仿這個(gè)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有沒(méi)有人知道dplyr等價(jià)物(？)或者是創(chuàng)建rleid行為與dplyr是執(zhí)行如下操作library(dplyr)my_rleid = rep(seq_along(rle(df$grp)$values), times = rle(df$grp)$lengths)df %>% mutate(rleid = my_rleid)

查看完整描述

3 回答

HUWWW

TA貢獻(xiàn)1874條經(jīng)驗(yàn) 獲得超12個(gè)贊

你可以做(當(dāng)你兩者兼?zhèn)涞臅r(shí)候)數(shù)據(jù)表和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

當(dāng)你不想裝載數(shù)據(jù)表另外，您還可以使用(如@DavidArenburg在注釋中提到的)：

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

你可以復(fù)制/竊取它：

myrleid <- data.table::rleid

反對(duì) 回復(fù) 2019-07-13

慕斯709654

TA貢獻(xiàn)1840條經(jīng)驗(yàn) 獲得超5個(gè)贊

如果你只想用R和dplyr，更好的方法是將您自己的一兩行版本的rleid()作為一個(gè)函數(shù)，然后在需要的時(shí)候應(yīng)用它。

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

反對(duì) 回復(fù) 2019-07-13

飲歌長(zhǎng)嘯

TA貢獻(xiàn)1951條經(jīng)驗(yàn) 獲得超3個(gè)贊

您可以使用lag功能來(lái)自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

反對(duì) 回復(fù) 2019-07-13