R разделить числовой вектор на позиции

Question

Я задаюсь вопросом о простой задаче разделения вектора на два в определенный индекс:

splitAt <- function(x, pos){ 
    list(x[1:pos-1], x[pos:length(x)]) 
} 

a <- c(1, 2, 2, 3) 

> splitAt(a, 4) 
[[1]] 
[1] 1 2 2 

[[2]] 
[1] 3 

Мой вопрос: Там должны быть некоторые существующие функции для этого, но я могу» Найти? Возможно, есть split? Моя наивная реализация также не работает, если pos=0 или pos>length(a).

Answer 1

Улучшение будет:

splitAt <- function(x, pos) unname(split(x, cumsum(seq_along(x) %in% pos))) 

который теперь может взять вектор позиций:

splitAt(a, c(2, 4)) 
# [[1]] 
# [1] 1 
# 
# [[2]] 
# [1] 2 2 
# 
# [[3]] 
# [1] 3 

И он ведет себя правильно (субъективно), если pos <= 0 или pos >= length(x) в том смысле, что он возвращает весь исходный вектор в одном элементе списка. Если вы хотите, чтобы вместо этого вышла ошибка, используйте stopifnot в верхней части функции.

Answer 2

Я пытался использовать flodel's answer, но это было слишком медленно в моем случае с очень большим x (и функция должна вызываться повторно). Поэтому я создал следующую функцию, которая намного быстрее, но также очень уродлива и не ведет себя правильно. В частности, он ничего не проверяет и вернет багги результаты, по крайней мере, для pos >= length(x) или pos <= 0 (вы можете добавить эти проверки самостоятельно, если вы не уверены в своих источниках и не слишком обеспокоены скоростью), и, возможно, и в некоторых других случаях , так что будь осторожен.

splitAt2 <- function(x, pos) { 
    out <- list() 
    pos2 <- c(1, pos, length(x)+1) 
    for (i in seq_along(pos2[-1])) { 
     out[[i]] <- x[pos2[i]:(pos2[i+1]-1)] 
    } 
    return(out) 
} 

Однако splitAt2 работает примерно в 20 раз быстрее, с й длиной 10 :

library(microbenchmark) 
W <- rnorm(1e6) 
splits <- cumsum(rep(1e5, 9)) 
tm <- microbenchmark(
        splitAt(W, splits), 
        splitAt2(W, splits), 
        times=10) 
tm 

Answer 3

Другой альтернативой, которая может быть быстрее и/или более удобными для чтения/элегантного, чем flodel's solution:

splitAt <- function(x, pos) { 
    unname(split(x, findInterval(x, pos))) 
}