R将前一值加5填充NA

发布于11月03日

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)，则取先前的值作为CONCENT，并向其添加5.但只能向下而不是向上.

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

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

推荐答案

您可以使用purrr::accumulate:

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中，有Reduce和accumulate = 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