计算相同长度向量中的连续元素 [英] Count consecutive elements in a same length vector

问题描述

如果我有像这样的向量

"a": 0 0 1 1 1 0 0 0 0 1 1 0 0 0

如何生成包含连续元素数的相同长度的向量，如下所示:

How can I generate a vector of the same length containing the count of consecutive elements, like so:

"b": 2 2 3 3 3 4 4 4 4 2 2 3 3 3

我尝试过rle，但是我没有设法以这种方式进行扩展.

I tried rle, but I did not manage to stretch it out this way.

推荐答案

使用rle和rep

with(rle(a), rep(lengths, times = lengths))
# [1] 2 2 3 3 3 4 4 4 4 2 2 3 3 3

数据

a <- c(0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0)

这篇关于计算相同长度向量中的连续元素的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！