通过使用 R 中的 lag() 添加到上面的行来填充列

Question

我想用不断添加到上面行的值填充现有列。 这在 Excel 中很容易，但我还没有找到在 R 中自动化它的好方法。

如果我们在 Excel 中有 2 列，A 和 B，我们希望单元格 B2 =B1+A2，单元格 B3 = B2 +A3。我怎样才能在 R 中做到这一点？

#example dataframe
df <- data.frame(A = 0:9, B = c(50,0,0,0,0,0,0,0,0,0))
#desired output
desired <- data.frame(A = 0:9, B = c("NA",51,53,56,60,65,71,78,86,95))

我尝试使用 lag() 函数，但它没有给出正确的输出。

df <- df %>%
mutate(B = B + lag(A))

所以我做了一个有效的 for 循环，但我觉得有更好的解决方案。

for(i in 2:nrow(df)){
df$B[i] <- df$B[i-1] + df$A[i]
}

最终，我想在整个数据帧的每 n 行上迭代此函数，本质上是让求和重置每 n 行。 （任何有关如何做到这一点的提示将不胜感激！）

Answer 1

这可能接近您的需要，并使用 tidyverse。具体来说，它使用来自 purrr 的 accumulate。

假设您想将每 n 行重置为零，您也可以提前使用 group_by。

目前尚不完全清楚您希望如何处理第一行；在这里，它只会使用第一个 B 值并忽略第一个 A 值，该值看起来与您在帖子中的内容类似。

n <- 5

library(tidyverse)

df %>%
  group_by(grp = ceiling(row_number() / n)) %>%
  mutate(B = accumulate(A[-1], sum, .init = B[1]))

输出

       A     B   grp
   <int> <dbl> <dbl>
 1     0    50     1
 2     1    51     1
 3     2    53     1
 4     3    56     1
 5     4    60     1
 6     5     0     2
 7     6     6     2
 8     7    13     2
 9     8    21     2
10     9    30     2

Answer 2

cumsum() 可用于获取您需要的结果。

df$B <- cumsum(df$B + df$A)
df
   A  B
1  0 50
2  1 51
3  2 53
4  3 56
5  4 60
6  5 65
7  6 71
8  7 78
9  8 86
10 9 95