R: znajdź najbliższy indeks

Question

Mam dwa wektory z kilku tysięcy punktów, ale uogólnione tutaj:

A <- c(10, 20, 30, 40, 50) 
b <- c(13, 17, 20) 

Jak mogę uzyskać indeksy z A że są najbliższy do b? Oczekiwany wynik to c(1, 2, 2).

wiem, że findInterval może znaleźć pierwsze wystąpienie, a nie najbliższa, i jestem świadomy, że which.min(abs(b[2] - A)) jest coraz cieplej, ale nie mogę dowiedzieć się, jak wektorować go do pracy z długimi wektorów zarówno A i b.

Answer 1

Możesz po prostu wpisać kod w sapply. Myślę, że ma tę samą prędkość, co pętla for, więc nie jest technicznie wektoryzowany:

sapply(b,function(x)which.min(abs(x - A))) 

Answer 2

FindInterval bardzo ci odpowiada. Po prostu trzeba wybierać między offset powraca i następny:

#returns the nearest occurence of x in vec 
nearest.vec <- function(x, vec) 
{ 
    smallCandidate <- findInterval(x, vec, all.inside=TRUE) 
    largeCandidate <- smallCandidate + 1 
    #nudge is TRUE if large candidate is nearer, FALSE otherwise 
    nudge <- 2 * x > vec[smallCandidate] + vec[largeCandidate] 
    return(smallCandidate + nudge) 
} 

nearest.vec(b,A) 

powroty (1,2,2), która powinna porównywalna FindInterval wydajności.

Answer 3

Oto rozwiązanie, które wykorzystuje często pomijaną funkcję R outer. Nie jestem pewien, czy to będzie lepsze, ale pozwoli uniknąć sapply.

A <- c(10, 20, 30, 40, 50) 
b <- c(13, 17, 20) 

dist <- abs(outer(A, b, '-')) 
result <- apply(dist, 2, which.min) 

# [1] 1 2 2