Вопрос по r – R: найти ближайший индекс

6

У меня есть два вектора с несколькими тысячами точек, но обобщенные здесь:

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

Как я могу получить признакиA которыеnearest вb? Ожидаемый результат будетc(1, 2, 2).

я знаю этоfindInterval может найти только первое вхождение, а не ближайшее, и я знаю, чтоwhich.min(abs(b[2] - A)) становится теплее, но я не могу понять, как векторизовать его для работы с длинными векторами обоихA а такжеb.

Ваш Ответ

3   ответа
11

что он имеет ту же скорость, что и цикл for, поэтому технически не векторизован:

sapply(b,function(x)which.min(abs(x - A)))
Обратите внимание, чтоwhich.min() возвращает только первое совпадение. Могут быть и другие элементы, которые одинаково близки. Sacha Epskamp
11

ду смещением, которое он возвращает, и следующим:

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

returns (1,2,2) и должны быть сопоставимы с FindInterval по производительности.

Действительно полезно, спасибо. Я удивлен, что в базе нечем заняться. Что, вероятно, означает, что есть, и я не знаю об этом ...! Flyto
0

Вот решение, которое использует R часто пропускаетсяouter функция. Не уверен, что это будет работать лучше, но он избегает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

Похожие вопросы