我如何在r data.frame中使用上一行中的值来优雅地计算一个变量

Question

这是我构建的简单场景：

说我有以下内容：

set.seed(1)
id<-sample(3,10,replace = TRUE)
n<-1:10
x<-round(runif(10,30,40))
df<-data.frame(id,n,x)
df
   id  n  x
1   1  1 32
2   2  2 32
3   2  3 37
4   3  4 34
5   1  5 38
6   3  6 35
7   3  7 37
8   2  8 40
9   2  9 34
10  1 10 38

我如何优雅地计算x.lag，其中x.lag是同一ID的先前x或0，如果不存在以前的值。

这是我所做的，但我对此不满意：

df$x.lag<-rep(0,10)
for (id in 1:3)
 df[df$id==id,]$x.lag<-c(0,df[df$id==id,]$x)[1:sum(df$id==id)]
df
   id  n  x x.lag
1   1  1 32     0
2   2  2 32     0
3   2  3 37    32
4   3  4 34     0
5   1  5 38    32
6   3  6 35    34
7   3  7 37    35
8   2  8 40    37
9   2  9 34    40
10  1 10 38    38

Answer 1

我们可以使用 data.table

library(data.table)
setDT(df)[, x.lag :=  shift(x, fill=0), id]

---

或使用dplyr

library(dplyr)
df %>%
  group_by(id) %>%
  mutate(x.lag = lag(x, default = 0))

---

或使用base R

的ave

df$x.lag <- with(df, ave(x, id, FUN = function(x) c(0, x[-length(x)])))
df$x.lag
#[1]  0  0 32  0 32 34 35 37 40 38