计算前 n 行的总和

Question

我有一个相当大的数据框，我正在尝试添加一个新变量，它是前三行在运行基础上的总和，它也应该按 ID 分组。每个 ID 的前三行应该是 0。它应该是这样的。

ID   Var1  VarNew
1     2      0
1     2      0
1     3      0
1     0      7
1     4      5
1     1      7

这是一个示例数据框

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

Answer 1

您可以使用任何具有滚动计算功能的软件包，window 大小为 3，结果 lag。例如 zoo::rollsumr.

library(dplyr)

df %>%
  group_by(ID) %>%
  mutate(VarNew = lag(zoo::rollsumr(Var1, 3, fill = 0), default = 0)) %>%
  ungroup

#     ID  Var1 VarNew
#  <dbl> <dbl>  <dbl>
#1     1     2      0
#2     1     2      0
33     1     3      0
#4     1     0      7
#5     1     4      5
#6     1     1      7

Answer 2

您可以在 ave 中使用 filter。

df$VarNew <- ave(df$Var1, df$ID, FUN=function(x) c(0, 0, 0,
 filter(head(df$Var1, -1), c(1,1,1), side=1)[-1:-2]))
df
#  ID Var1 VarNew
#1  1    2      0
#2  1    2      0
#3  1    3      0
#4  1    0      7
#5  1    4      5
#6  1    1      7

或将 cumsum 与 head 和 tail 组合使用。

df$VarNew <- ave(df$Var1, df$ID, FUN=function(x) {y <- cumsum(x)
  c(0, 0, 0, tail(y, -3) - head(y, -3))})

Answer 3

图书馆runner也有帮助

library(runner)
df %>% mutate(var_new = sum_run(Var1, k =3, na_pad = T, lag = 1))

  ID Var1 var_new
1  1    2      NA
2  1    2      NA
3  1    3      NA
4  1    0       7
5  1    4       5
6  1    1       7

如果需要，

NAs 可以很容易地突变为 0。

计算前 n 行的总和

Calculate sum of n previous rows

r

sum

dataframe

dplyr

rolling-computation