rollapply na.rm = TRUE给出0值，而不是NA

Question

我有一个简单的问题，我似乎无法在google，stackoverflow或stackexchange上找到答案。 我目前正在使用rollapply的示例来查找包含NA的某些值的总和。 例如：

 z <- zoo(c(NA, NA, NA, NA,2, 3, 4, 5, NA))
 rollapply(z, 3, sum, na.rm = TRUE, align = "right")

输出：

  3  4  5  6  7  8  9 
  0  0  2  5  9 12  9 

看起来不错，但是有两次连续有3个NA。 和功能将NA交换为0。 不幸的是，这不适用于我要处理的数据，因为0是有意义的值。 有没有办法用NA再次替换0？

我正在寻找如下输出：

  3   4   5  6  7  8  9 
  NA  NA  2  5  9 12  9 

先感谢您！

Answer 1

您可以执行以下操作（即使不是很好）

require(zoo)
z <- zoo(c(NA, NA, NA, NA,2, 3, 4, 5, NA))
tmp <- rollapply(z, 3, sum, na.rm = TRUE, align = "right")

tmp[is.na(z)[-2:-1] & tmp == 0] <- NA
tmp

因此，在z为na的地方分配NA ，而rollapply产生NA

这给你：

> tmp
 3  4  5  6  7  8  9 
NA NA  2  5  9 12  9 

Answer 2

这是两种方法：

1）请注意，并不是rollapply给出0，而是sum(x, na.rm = TRUE)给出0。这里的功能sum(x, na.rm = TRUE)并不是真正想要的。

而是提供一个以所需方式工作的sum版本，即，当输入完全为NA值时返回NA sum(x, na.rm = TRUE)否则返回sum(x, na.rm = TRUE) 。

sum_na <- function(x) if (all(is.na(x))) NA else sum(x, na.rm = TRUE)
rollapplyr(z, 3, sum_na)

2）或者，使用你的代码，并通过更换其输入是所有的任何位置之后修复它NA用NA ：

zz <- rollapplyr(z, 3, sum, na.rm = TRUE)
zz[rollapply(is.na(z), 3, all)] <- NA

赠送：

> zz
 3  4  5  6  7  8  9 
NA NA  2  5  9 12  9