R podzielić wektor numerycznej w pozycji

Question

Zastanawiam o prostym zadaniem dzielenie wektora na dwie części w określonym indeksie:

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 

Moje pytanie: Tam musi być jakaś istniejąca funkcja dla tego, ale mogę” t go znaleźć? Czy możliwe jest, że split? Moja naiwna implementacja również nie działa, jeśli pos=0 lub pos>length(a).

Answer 1

Poprawa będzie:

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

który może teraz podjąć wektor pozycji:

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

I to nie zachowywać prawidłowo (subiektywna) jeżeli pos <= 0 lub pos >= length(x) w tym sensie, że zwraca cały oryginalny wektor w pojedynczym elemencie listy. Jeśli zamiast tego chcesz, aby błąd się pojawił, użyj funkcji stopifnot u góry funkcji.

Answer 2

Próbowałem użyć flodel's answer, ale w moim przypadku było zbyt wolno z bardzo dużym x (a funkcja musi być wielokrotnie wywoływana). Stworzyłem więc następującą funkcję, która jest znacznie szybsza, ale również bardzo brzydka i nie zachowuje się właściwie. W szczególności nie sprawdza niczego i zwróci wyniki błędne co najmniej dla pos >= length(x) lub pos <= 0 (możesz dodać te sprawdzenia samodzielnie, jeśli nie masz pewności co do swoich danych wejściowych i nie przejmujesz się zbytnio szybkością), a może również inne przypadki , więc uważaj.

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) 
} 

Jednakże splitAt2 przebiega około 20 razy szybciej z X o długości 10 :

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

Answer 3

Inna alternatywa może być szybszy i/lub bardziej czytelny/eleganckim niż flodel's solution:

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