如果在 2 行中滿足某些條件，如何在 R 數據框中添加新列，顯示當前行和前一行中的值之和？

Question

假設您有一個由“a”和“b”列組成的數據框，其值如下所示，由df <- data.frame(a=c(0, 1, 2, 2, 3), b=c(1, 3, 8, 9, 4)) 。 假設您要添加“c”列，如果“a”中的值等於“a”列中前一行的值，則將“b”列中的相應行值相加； 否則顯示 0 值。 下面添加了一個“c”列來說明我正在嘗試做的事情：

   a  b   add col c
1  0  1       0
2  1  3       0
3  2  8       0
4  2  9       17 (since the values in col "a" rows 3 and 4 are equal, add the values in col b rows 3 and 4)
5  3  4       0

在本機 R 中執行此操作的最簡單方法是什么？

Answer 1

我們可以用

 df %>% group_by(a) %>% mutate(c = if(n() > 1) sum(b) else 0)