R 将 NA 与上一个值相加 5 来填充 NA-解网

问：

HAVE = data.frame(STUDENT = c(1,1,1,2,2,2,3,3),
                  TIME = c(1,2,3,1,2,3,1,2),
                  SCORE = c(7, NA, NA, 5, NA, 19, NA, 2))

WANT = data.frame(STUDENT = c(1,1,1,2,2,2,3,3),
                  TIME = c(1,2,3,1,2,3,1,2),
                  SCORE = c(7, 12, 17, 5, 10, 19, NA, 2))

我希望通过执行以下步骤来修改 SCORE：如果 is.na（SCORE），则取 STUDENT 的前一个值并加 5。但只能向下做，而不是向上。

我尝试过这个没有成功，当学生连续有两个 NA 时它不起作用

HAVE %>% group_by(STUDENT) %>% mutate(WANT = ifelse(is.na(SCORE), lag(SCORE) + 5, SCORE))

r dplyr na

library(dplyr)
library(purrr)
HAVE |> 
  mutate(SCORE = accumulate(SCORE, ~ if(is.na(.y)) .x + 5 else .y),
         .by = STUDENT)

#   STUDENT TIME SCORE
# 1       1    1     7
# 2       1    2    12
# 3       1    3    17
# 4       2    1     5
# 5       2    2    10
# 6       2    3    19
# 7       3    1    NA
# 8       3    2     2

或者，在基数 R 中，使用和：Reduceaccumulate = TRUE

na_add <- function(col, n = 5){
  Reduce(function(x, y) if(is.na(y)) x + n else y, col, accumulate = TRUE)
}
with(HAVE, ave(SCORE, STUDENT, FUN = na_add))
#[1]  7 12 17  5 10 19 NA  2

上一个：如何将特定列中的所有 NA 重命名为 0，然后在整个数据帧中删除这些行（0s）？

下一个：按组 [duplicate] 删除 NA

R 将 NA 与上一个值相加 5 来填充 NA

R Fill NA With Adding 5 to Previous Value

评论