在底部的 MWE 代碼中，我試圖在從一行運行到下一行時為每個唯一 id 生成運行余額。例如，運行以下代碼時，輸??出應為：
data2 <-
id plusA plusB minusC running_balance [desired calculation for running balance]
1 3 5 10 -2 3 5 - 10 = -2
2 4 5 9 0 4 5 - 9 = 0
3 8 5 8 5 8 5 - 8 = 5
3 1 4 7 3 id doesn't change so 5 from above (1 4 - 7) = 3
3 2 5 6 4 id doesn't change so 3 from above (2 5 - 6) = 4
5 3 6 5 4 3 6 - 5 = 4
下面的MWE是指，當id從一行到下一行一致時，前一行plusA數量而不是前一行running_balance數量。我已經嘗試將下面的更改為某種形式，lag(running_balance...)但還沒有運氣。
我試圖盡量減少使用太多包。例如，我知道該purrr包提供了一個accumulate()功能，但我只想暫時堅持dplyr。有沒有一種簡單的方法可以做到這一點，mutate()在我的情況下使用 dplyr ？我還嘗試擺弄cumsum()應該在這里作業的 dplyr函式，但我不確定如何將它們中的幾個串在一起。
MWE代碼：
data <- data.frame(id=c(1,2,3,3,3,5),
plusA=c(3,4,8,1,2,3),
plusB=c(5,5,5,4,5,6),
minusC = c(10,9,8,7,6,5))
library(dplyr)
data2<- subset(
data %>% mutate(extra=case_when(id==lag(id) ~ lag(plusA), TRUE ~ 0)) %>%
mutate(running_balance=plusA plusB-minusC extra),
select = -c(extra)
)
uj5u.com熱心網友回復：
使用dplyr：
data %>%
mutate(running_balance = plusA plusB - minusC) %>%
group_by(id) %>%
mutate(running_balance = cumsum(running_balance)) %>%
ungroup()
輸出：
# A tibble: 6 x 5
# Groups: id [4]
id plusA plusB minusC running_balance
<dbl> <dbl> <dbl> <dbl> <dbl>
1 1 3 5 10 -2
2 2 4 5 9 0
3 3 8 5 8 5
4 3 1 4 7 3
5 3 2 5 6 4
6 5 3 6 5 4
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/388495.html
上一篇：jupyter-notebook中r語言的資料顯示問題
下一篇：R函式在每一行增加數字