R中( 0，1，0，0，1，1，1)到(0，0，0，1，0，1，2)的转换

Question

我认为这是一个相当基本的问题，但我想看看是否有人能想出一个更优雅的解决方案。也许是通过避免for循环。

我想要一个函数，它以1和0的向量作为输入，并返回一个长度相同的向量，它计算相同数目的前几个位置。

这样做的一个相当不雅的方法是：

count_me <- function(x) {
  count_vector <- numeric(length(x))
  for(i in 2:length(x)) {
    if(x[i] == x[i-1]) count_vector[i] <- count_vector[i-1] + 1
  }
  count_vector
}

这正是我想要的：

> (p <- sample(c(0,1), size = 10, replace = TRUE))
 [1] 0 1 0 1 1 0 0 1 1 1
> count_me(p)
 [1] 0 0 0 0 1 0 1 0 1 2

Answer 1

我认为这样做是可行的：

sequence(rle(x)$lengths) - 1