r - 如何从向量或列表中消除 n 个连续值？

Question

假设我有:

v <- c(1,2,0,0,0,3,4,4,4,0,0,0,0,5,0)
v
 [1] 1 2 0 0 0 3 4 4 4 0 0 0 0 5 0

我想消除连续 0 的数量为 2+(2 或更多)的任何连续 0。所以上面的例子会变成:

1 2 3 4 4 4 5 0

NOTE: I would prefer a base-R solution but other solutions would be interesting

Answer 1

1) na.locf0 这个使用zoo包但是只有一行代码。它利用 na.locf0 中的 maxgap 参数。用 NA 替换每个 0，用 0 替换其他所有内容。使用 na.locf0 和 maxgap 来填充它，将其添加到原始向量中，然后应用 na。省略来移除NAs并使用c来移除属性。

library(zoo)

c(na.omit(na.locf0(ifelse(v == 0, NA, 0), maxgap = 1) + v))
## [1] 1 2 3 4 4 4 5 0

2) rleid 这使用了 data.table 中的 rleid。它比(1)长一点，但仍然相当短。它使用 rleid 对数据进行分组，然后为每个组生成一个 NA 或在最后删除 NA 的原始数据。

library(data.table)

fun <- function(x) if (x[1] == 0 & length(x) > 1) NA else x
c(na.omit(ave(v, rleid(v), FUN = fun)))
## [1] 1 2 3 4 4 4 5 0