Question

我的问题很简单，但我仍然无法在线找到解决方案。

我有一个载体，例如a = c(7,3,5)。我需要将其转换为b = c(0,1,2,3,4,5,6,0,1,2,0,1,2,3,4)。我可以通过一个循环来做到这一点，但是当长度（a）＆gt;时它出了名的慢。 500000。

m <- 0
n <- 0
for (i in 1:length(a)) {
  m <- n+1;
  n <- n+a[i];

  b[m:n] <- (0:(a[i]-1));
}

R中是否存在可以快速产生所描述行为的单线程？同样的方法可以将向量a转换为c = c(0,0,0,0,0,0,1,0,0,1,0,0,0,0,1)吗？

Answer 1

评论中有几个选项，但在答案区域中没有发布，所以在寻找可能的重复版的同时，这里是我们迄今为止的整合。首先列出每个选项的最直接/逻辑替代方案。

## To get your first vector
sequence(a) - 1                                              # @Henrik
ave(1:sum(a), rep(seq_along(a), a), FUN = seq_along) - 1     # @akrun

## To get your second vector
tabulate(cumsum(a))                                          # @alexis_laz
{ x <- integer(sum(a)) ; x[cumsum(a)] <- 1; x }              # @DavidArenburg
{ x <- sequence(a) - 1 ; as.integer(c(diff(x) != 1, TRUE)) } # @Henrik
sequence(a) %/% rep(a, a)                                    # @GL_Li

这个答案是社区维基，所以随时编辑和添加替代品。

R中不同长度的序列

1 个答案: