移的可变向上在R数据帧相对于另一变量

Question

我有一个data.frame：

ID <-c(2,2,2,2,3,3,5,5) 
Pur<-c(0,1,2,3,1,2,4,5) 
df<-data.frame(ID,Pur) 

我想推Pur为每个ID得到up.Pur如下：

ID Pur up.Pur 
2 0 1 
2 1 2 
2 2 3 
2 3 NA 
3 1 2 
3 2 NA 
5 4 5 
5 5 NA 

希望你对此有所帮助。

Answer 1

我们可以使用shift从data.table

library(data.table) 
setDT(df)[, up.Pur := shift(Pur, type = "lead"), by = ID] 
df 
# ID Pur up.Pur 
#1: 2 0  1 
#2: 2 1  2 
#3: 2 2  3 
#4: 2 3  NA 
#5: 3 1  2 
#6: 3 2  NA 
#7: 5 4  5 
#8: 5 5  NA 

Answer 2

这里是一个dplyr方法

library(dplyr) 

ID <-c(2,2,2,2,3,3,5,5) 
Pur<-c(0,1,2,3,1,2,4,5) 
df<-data.frame(ID,Pur) 

df %>% 
    group_by(ID) %>% 
    mutate(up.Pur = lead(Pur)) 

# Source: local data frame [8 x 3] 
# Groups: ID [3] 
# 
#  ID Pur up.Pur 
# <dbl> <dbl> <dbl> 
# 1  2  0  1 
# 2  2  1  2 
# 3  2  2  3 
# 4  2  3  NA 
# 5  3  1  2 
# 6  3  2  NA 
# 7  5  4  5 
# 8  5  5  NA 

为了完整起见，我添加了一个基础R的做法，万一你不觉得像安装任何软件包。

dfList = split(df, ID) 

dfList = lapply(dfList, function(x){ 
    x$up.Pur = c(x$Pur[-1], NA) 
    return(x) 
}) 

unsplit(dfList, ID) 

# ID Pur up.Pur 
# 1 2 0  1 
# 2 2 1  2 
# 3 2 2  3 
# 4 2 3  NA 
# 5 3 1  2 
# 6 3 2  NA 
# 7 5 4  5 
# 8 5 5  NA