如果 NA 出现在向量中,则将所有后续元素设为 NA

If NA appears in a vector, make all subsequent elements NA

一旦初始 NA 出现在向量中,我想用 NA 替换该向量中的所有后续值。

例如,如果x = c(0, 0, NA, 0, 0, 1, 2)

我想要:x = c(0, 0, NA, NA, NA, NA, NA)

如何操作?

可以做到:

x <- ifelse(cumsum(is.na(x)) > 0, NA, x)

或更快:

x[cumsum(is.na(x)) > 0] <- NA

输出:

[1]  0  0 NA NA NA NA NA
v <- c(0,0, NA, 0, 0, 1, 2)
v[ as.logical(cummax(is.na(v))) ] <- NA
v
# [1]  0  0 NA NA NA NA NA

也可以使用

v[ dplyr::cumany(is.na(v)) ] <- NA