在新列中连接当前行和后续行

Question

假设我们在R中有这个数据框:

df <- data.frame(id = c(rep(1,5), rep(2, 3), rep(3, 4), rep(4, 2)), brand = c("A", "B", "A", "D", "Closed", "B", "C", "D", "D", "A", "B", "Closed", "C", "Closed"))

> df
#   id  brand
#1   1      A
#2   1      B
#3   1      A
#4   1      D
#5   1 Closed
#6   2      B
#7   2      C
#8   2      D
#9   3      D
#10  3      A
#11  3      B
#12  3 Closed
#13  4      C
#14  4 Closed

我想创建一个新变量来表示从当前行到下一行的品牌列中的更改,但这必须仅在每个ID号内发生.

我创建了新列:

df$brand_chg <- ""

这个循环正确地完成了我想要做的事情:

for (i in 1:nrow(df)) {

    j <- i + 1

    if(j > nrow(df)) next #to prevent error in very last row

    if (df[i,'id'] != df[j, 'id']) next #to skip loop when id changes

    df[i,'brand_chg'] <- paste(df[i,'brand'], df[j,'brand'], sep = "->") 
    #populating concatenation
}

#Results:
#   id  brand brand_chg
#1   1      A      A->B
#2   1      B      B->A
#3   1      A      A->D
#4   1      D D->Closed
#5   1 Closed          
#6   2      B      B->C
#7   2      C      C->D
#8   2      D          
#9   3      D      D->A
#10  3      A      A->B
#11  3      B B->Closed
#12  3 Closed          
#13  4      C C->Closed
#14  4 Closed 

但是,在具有287k行的表上,此循环至少需要10分钟才能运行.有谁知道更快的方法来实现这种连接？

谢谢,我感谢您的见解.

Answer 1

使用dplyr包:

library(dplyr)

df %>% group_by(id) %>% 
    mutate(brand_chg = ifelse(seq_along(brand) == n(), 
                              "", 
                              paste(brand, lead(brand), sep = "->")))