如何在滿足條件時將向量拆分為向量列表？

Question

我想將一個向量拆分成一個向量列表。 生成的向量將具有可變長度，我需要僅在滿足特定條件時才進行拆分。

樣本數據：

set.seed(3)
x <- sample(0:9,100,repl=TRUE)

例如，在這種情況下，我想在每個 0 處拆分上述向量x 。

目前我用我自己的 function 這樣做：

ConditionalSplit <- function(myvec, splitfun) {
  newlist <- list()
  splits <- which(splitfun(x))
  if (splits == integer(0)) return(list(myvec))
  if (splits[1] != 1) newlist[[1]] <- myvec[1:(splits[1]-1)]
  i <- 1
  imax <- length(splits)

  while (i < imax) {
    curstart <- splits[i]
    curend <- splits[i+1]
    if (curstart != curend - 1)
      newlist <- c(newlist, list(myvec[curstart:(curend-1)]))
    i <- i + 1
  }

  newlist <- c(newlist, list(myvec[splits[i]:length(vector)]))
  return(newlist)
}

這個 function 給出了我想要的 output，但我確信有比我的更好的方法。

> MySplit <- function(x) x == 0

> ConditionalSplit(x, MySplit)

[[1]]
 [1] 1 8 3 3 6 6 1 2 5 6 5 5 5 5 8 8 1 7 8 2 2

[[2]]
[1] 0 1

[[3]]
 [1] 0 2 7 5 9 5 7 3 3 1 4 2 3 8 2 5 2 2 7 1 5 4 2
...

Answer 1

以下行似乎工作得很好：

split(x,cumsum(x==0))

Answer 2

另一種解決方案是使用 tapply。 使用 tapply 而不是 split 的一個很好的理由是因為它允許您在拆分列表時對列表中的項目執行其他操作。

例如，在這個問題的解決方案中：

> x <- sample(0:9,100,repl=TRUE)
> idx <- cumsum(x==0)
> splitList <- tapply(x, idx, function(y) {list(y)})
> splitList
$`0`
[1] 2 9 2

$`1`
[1] 0 5 5 3 8 4

$`2`
[1] 0 2 5 2 6 2 2

$`3`
[1] 0 8 1 7 5

$`4`
 [1] 0 1 6 6 3 8 7 2 4 2 3 1

$`5`
[1] 0 6 8 9 9 1 1 2

$`6`
 [1] 0 1 2 2 2 7 8 1 9 7 9 3 4 8 4 6 4 5 3 1

$`7`
[1] 0 2 7 8 5

$`8`
[1] 0 3 4 8 4 7 3

$`9`
[1] 0 8 4

$`10`
 [1] 0 4 3 9 9 8 7 4 4 5 5 1 1 7 3 9 7 4 4 7 7 6 3 3

可以修改，以便將每個元素除以該列表中的元素數。

list(y/length(y))

代替

list(y)