获取向量中缺失值的连续长度

Question

有没有一种巧妙的（即不使用循环）方法来获取向量中每个缺失值序列的长度？ 我理想的输出是一个相同长度的向量，其中每个缺失值被替换为其所属的缺失值序列的长度，所有其他值都是0。

因此，对于像这样的输入：

x <- c(2,6,1,2,NA,NA,NA,3,4,NA,NA)

我希望能够得到以下输出：

y <- c(0,0,0,0,3,3,3,0,0,2,2)

- daanoo

Answer 1

使用rle的一个简单选项：

m <- rle(is.na(x))
> rep(ifelse(m$values,m$lengths,0),times = m$lengths)
[1] 0 0 0 0 3 3 3 0 0 2 2

- joran

---

rep(rle(is.na(x))$value * rle(is.na(x))$length, rle(is.na(x))$length)。这也是有效的。 - JasonWang

Answer 2

我正在独立地使用 rle() 和 cumsum() 或 dplyr group_by() 以及 n() 来获取NA的组长度：

> x2 <- as.numeric(is.na(x))
  0 0 0 0 1 1 1 0 0 1 1

> rle(x2)
Run Length Encoding
  lengths: int [1:4] 4 3 2 2
  values : num [1:4] 0 1 0 1

# Now we can assign group-numbers...
> cumsum(c(diff(x2)==+1,0)) * x2
  0 0 0 0 1 1 1 0 0 2 2
# ...then get group-lengths from counting those...
> rle(cumsum(c(diff(x2)==+1,0)) * x2)
Run Length Encoding
  lengths: int [1:4] 4 3 2 2
  values : num [1:4] 0 1 0 2

我们可以凑合着做点什么，但它不会像@joran的解决方案那样紧凑和优雅。

- smci

Answer 3

这里有另一种选项，使用rleid和ave

library(data.table)
ave(x, rleid(is.na(x)), FUN = length)*is.na(x)
#[1] 0 0 0 0 3 3 3 0 0 2 2

- akrun