使用dplyr按組獲取累積計數

Question

提前致謝。 我有以下數據：

df <- data.frame(person=c(1,1,1,1,2,2,2,2,3,3,3,3), 
             neighborhood=c("A","A","A","A","B","B","C","C","D","D","E","F"))

我想生成一個新列，它給出了每個人在面板進行過程中所經過的鄰域的累積計數。 像這樣：

df2 <- data.frame(person=c(1,1,1,1,2,2,2,2,3,3,3,3), 
             neighborhood=c("A","A","A","A","B","B","C","C","D","D","E","F"),
             moved=c(0,0,0,0,0,0,1,1,0,0,1,2)
             )

再次感謝。

Answer 1

我們可以使用'person'分組，然后通過將'鄰域'與其unique值match來創建'移動'以獲得索引並減去1。

df %>%
   group_by(person) %>% 
   mutate(moved = match(neighborhood, unique(neighborhood))-1)
#   person neighborhood moved
#    <dbl>       <fctr> <dbl>
#1       1            A     0
#2       1            A     0
#3       1            A     0
#4       1            A     0
#5       2            B     0
#6       2            B     0
#7       2            C     1
#8       2            C     1
#9       3            D     0
#10      3            D     0
#11      3            E     1
#12      3            F     2

或者使用具有指定為'鄰域'中unique值的levels factor ，強制為'整數'並減去1。

df %>%
   group_by(person) %>% 
   mutate(moved = as.integer(factor(neighborhood, levels = unique(neighborhood)))-1)
#   person neighborhood moved
#    <dbl>       <fctr> <dbl>
#1       1            A     0
#2       1            A     0
#3       1            A     0
#4       1            A     0
#5       2            B     0
#6       2            B     0
#7       2            C     1
#8       2            C     1
#9       3            D     0
#10      3            D     0
#11      3            E     1
#12      3            F     2

Answer 2

使用data.table包中的rleid或frank函數也可以輕松實現這data.table ：

library(data.table)
# with 'rleid'
setDT(df)[, moved := rleid(neighborhood)-1, by = person]
# with 'frank'
setDT(df)[, moved := frank(neighborhood, ties.method='dense')-1, by = person]

結果：

> df
    person neighborhood moved
 1:      1            A     0
 2:      1            A     0
 3:      1            A     0
 4:      1            A     0
 5:      2            B     0
 6:      2            B     0
 7:      2            C     1
 8:      2            C     1
 9:      3            D     0
10:      3            D     0
11:      3            E     1
12:      3            F     2

使用dplyr您可以使用dense_rank函數：

library(dplyr)
df %>%
  group_by(person) %>%
  mutate(moved = dense_rank(neighborhood)-1)

Answer 3

這也可以使用dplyr窗口函數來實現。 這是代碼：

library(dplyr)
my.df <- tbl_df(df)

my.df %>% 
    # Per person
    group_by(person) %>% 
    # sort by neighborhood
    arrange(neighborhood) %>%
    # if the neighborhood has changed compared to the row before
    mutate(moved = (neighborhood != lag(neighborhood))) %>%
    # turn NAs (first rows) into FALSE
    mutate(moved = ifelse(is.na(moved), FALSE, moved)) %>%
    # use cumulative sum of the logical column to get number of moves
    mutate(no_moves = cumsum(moved))