rollapply返回NA？

Question

x <- 1:10
x[6] <- NA
x
#[1]  1  2  3  4  5 NA  7  8  9 10
rollapply(x, 3, mean)
#[1] 2  3  4 NA NA NA  8  9

如果移动窗口中有超过3个NA，我希望rollapply返回NA。 有可能吗？

Answer 1

也许再次使用rollapply查找包含三个或更多NA的索引，并用NA替换这些索引的值

x = c(1L, 2L, 3L, 4L, NA, NA, NA, 8L, 9L, 10L)

library(zoo)
replace(x = rollapply(x, 3, mean, na.rm = TRUE),
       list = which(rollapply(data = is.na(x), 3, sum) >= 3),
       values = NA)
#[1] 2.0 3.0 3.5 4.0  NA 8.0 8.5 9.0

Answer 2

您可以通过更改function中的rollapply参数直接执行此操作。类似的东西：

#Same input as d.b
x = c(1L, 2L, 3L, 4L, NA, NA, NA, 8L, 9L, 10L)

rollapply(x, 3, function(x) ifelse(sum(is.na(x))>=3, NA, mean(x, na.rm=T)))
#[1] 2.0 3.0 3.5 4.0  NA 8.0 8.5 9.0