r - Нахождение ближайших координат между двумя большими наборами данных - PullRequest
Новая
       55

r - Нахождение ближайших координат между двумя большими наборами данных

1 голос
/ 18 апреля 2019

Я стремлюсь определить ближайшую запись в наборе данных 2 для каждой записи в наборе данных 1 на основе координат в обоих наборах данных.Набор данных 1 содержит 180 000 строк (всего 1800 уникальных координат), а набор данных 2 содержит 4500 строк (полных 4500 уникальных координат).

Я попытался повторить ответы на аналогичные вопросы по stackoverflow.например:

R - Нахождение ближайшей соседней точки и количества соседей в пределах данного радиуса, координаты широта-долгота

Расчет расстояния между точками в разныхфреймы данных

Однако они не решают проблему так, как я хочу (они либо присоединяются к фреймам данных, либо проверяют расстояния в одном фрейме данных).

Решение в Найдите ближайшую координату X, Y, используя R и связанные посты являются ближайшими, которые я нашел до сих пор.

Моя проблема с постом в том, что он определяет расстояниемежду координатами в одном кадре данных, и я не смог понять, какие параметры нужно изменить в RANN::nn2, чтобы сделать это в двух кадрах данных.

Предлагаемый код, который не работает: 

library(RANN)
dataset1[,4]<- nn2(data=dataset1, query=dataset2, k=2)

Примечания / Вопросы:

1) Какой набор данных следует предоставить запросу, чтобы найти наиболее близкое значение в наборе данных 2 к заданному значению в наборе данных 1?

2) Есть ли какие-либопутьчтобы избежать проблемы, когда наборы данных должны иметь одинаковую ширину (количество столбцов)?

3) Как можно добавить выходные данные (SRD_ID и distance) к соответствующей записи в наборе данных1?

4) Как использовать параметр eps в функции RANN::nn2?

Цель состоит в том, чтобы заполнить столбцы SRC_ID и distance в наборе данных 1идентификатор ближайшей станции из набора данных 2 и расстояние между записью в наборе данных 1 и ближайшей записью в наборе данных 2.

Ниже приведена таблица, демонстрирующая ожидаемые результаты. Примечание: значения SRC_ID и distance являются примерами значений, которые я сам добавил вручную, почти наверняка неверны и, скорее всего, не будут реплицированы кодом. 

       id HIGH_PRCN_LAT HIGH_PRCN_LON SRC_ID distance
1 3797987      52.88121     -2.873734     55      350 
2 3798045      53.80945     -2.439163     76     2100

Данные:

r подробности 

platform        x86_64-w64-mingw32
version.string  R version 3.5.3 (2019-03-11)

набор данных 1 вход (не сужен до уникальных координат) 

structure(list(id = c(1L, 2L, 4L, 5L, 
6L, 7L, 8L, 9, 10L, 3L), 
    HIGH_PRCN_LAT = c(52.881442267773, 57.8094538200198, 34.0233529, 
    63.8087900198, 53.6888144440184, 63.4462810678651, 21.6075544376207, 
    78.324442654172, 66.85532539759495, 51.623544596), HIGH_PRCN_LON = c(-2.87377812157822, 
    -2.23454414781635, -3.0984448341, -2.439163178635, -7.396111601421454, 
    -5.162345043546359, -8.63311254098095, 3.813289888829932, 
    -3.994325961186105, -8.9065532453272409), SRC_ID = c(NA, NA, 
    NA, NA, NA, NA, NA, NA, NA, NA), distance = c(NA, NA, 
    NA, NA, NA, NA, NA, NA, NA, NA)), row.names = c(NA, 10L), class = "data.frame")

набор данных 2 ввода 

structure(list(SRC_ID = c(55L, 54L, 23L, 11L, 44L, 21L, 76L, 
5688L, 440L, 61114L), HIGH_PRCN_LAT = c(68.46506, 50.34127, 61.16432, 
42.57807, 52.29879, 68.52132, 87.83912, 55.67825, 29.74444, 34.33228
), HIGH_PRCN_LON = c(-5.0584, -5.95506, -5.75546, -5.47801, -3.42062, 
-6.99441, -2.63457, -2.63057, -7.52216, -1.65532)), row.names = c(NA, 
10L), class = "data.frame")

Ответы [ 2 ]

1 голос
/ 18 апреля 2019

Данные 

x = structure(list(id = c(1L, 2L, 4L, 5L, 
6L, 7L, 8L, 9, 10L, 3L), 
    HIGH_PRCN_LAT = c(52.881442267773, 57.8094538200198, 34.0233529, 
    63.8087900198, 53.6888144440184, 63.4462810678651, 21.6075544376207, 
    78.324442654172, 66.85532539759495, 51.623544596), HIGH_PRCN_LON = c(-2.87377812157822, 
    -2.23454414781635, -3.0984448341, -2.439163178635, -7.396111601421454, 
    -5.162345043546359, -8.63311254098095, 3.813289888829932, 
    -3.994325961186105, -8.9065532453272409), SRC_ID = c(NA, NA, 
    NA, NA, NA, NA, NA, NA, NA, NA), distance = c(NA, NA, 
    NA, NA, NA, NA, NA, NA, NA, NA)), row.names = c(NA, 10L), class = "data.frame")

y = structure(list(SRC_ID = c(55L, 54L, 23L, 11L, 44L, 21L, 76L, 
 5688L, 440L, 61114L), HIGH_PRCN_LAT = c(68.46506, 50.34127, 61.16432, 
 42.57807, 52.29879, 68.52132, 87.83912, 55.67825, 29.74444, 34.33228
 ), HIGH_PRCN_LON = c(-5.0584, -5.95506, -5.75546, -5.47801, -3.42062, 
 -6.99441, -2.63457, -2.63057, -7.52216, -1.65532)), row.names = c(NA, 
 10L), class = "data.frame")

Решение.Обратите внимание на «3: 2», чтобы получить «долготу / широту», в этом порядке. 

library(raster)

d <- pointDistance(x[,3:2], y[,3:2], lonlat=TRUE, allpairs=T) 
i <- apply(d, 1, which.min)

x$SRC_ID = y$SRC_ID[i]
x$distance = d[cbind(1:nrow(d), i)]
x

#   id HIGH_PRCN_LAT HIGH_PRCN_LON SRC_ID   distance
#1   1      52.88144     -2.873778     44   74680.48
#2   2      57.80945     -2.234544   5688  238553.51
#3   4      34.02335     -3.098445  61114  137385.18
#4   5      63.80879     -2.439163     23  340642.70
#5   6      53.68881     -7.396112     44  308458.73
#6   7      63.44628     -5.162345     23  256176.88
#7   8      21.60755     -8.633113    440  908292.28
#8   9      78.32444      3.813290     76 1064419.47
#9  10      66.85533     -3.994326     55  185119.29
#10  3      51.62354     -8.906553     54  251580.45

Иллюстрированный 

plot(x[,3:2], ylim=c(0,90), col="blue", pch=20)
points(y[,3:2], col="red", pch=20)
for (i in 1:nrow(x)) {
    j <- y$SRC_ID==x$SRC_ID[i]
    arrows(x[i,3], x[i,2], y[j,3], y[j,2],length=.1) 
}

text(x[,3:2], labels=x$id, pos=1, cex=.75)
text(y[,3:2], labels=y$SRC_ID, pos=3, cex=.75)

plot

1 голос
/ 18 апреля 2019

Я написал ответ, ссылаясь на эту тему . Функция модифицирована, чтобы заботиться о сообщении расстояния и избежать жесткого кодирования. Обратите внимание, что он рассчитывает евклидово расстояние . 

library(data.table)
#Euclidean distance 
mydist <- function(a, b, df1, x, y){

          dt <- data.table(sqrt((df1[[x]]-a)^2 + (df1[[y]]-b)^2))

          return(data.table(Closest.V1  = which.min(dt$V1),
                            Distance    = dt[which.min(dt$V1)]))
           }

setDT(df1)[, j = mydist(HIGH_PRCN_LAT, HIGH_PRCN_LON, setDT(df2), 
                        "HIGH_PRCN_LAT", "HIGH_PRCN_LON"), 
                         by = list(id, HIGH_PRCN_LAT, HIGH_PRCN_LON)]

  #     id HIGH_PRCN_LAT HIGH_PRCN_LON Closest.V1 Distance.V1
  # 1:   1      52.88144     -2.873778          5   0.7990743
  # 2:   2      57.80945     -2.234544          8   2.1676868
  # 3:   4      34.02335     -3.098445         10   1.4758202
  # 4:   5      63.80879     -2.439163          3   4.2415854
  # 5:   6      53.68881     -7.396112          2   3.6445416
  # 6:   7      63.44628     -5.162345          3   2.3577811
  # 7:   8      21.60755     -8.633113          9   8.2123762
  # 8:   9      78.32444      3.813290          7  11.4936496
  # 9:  10      66.85533     -3.994326          1   1.9296370
  # 10:  3      51.62354     -8.906553          2   3.2180026

Вы можете использовать RANN::nn2, но вам нужно убедиться, что вы используете правильный синтаксис. Следующие работы! 

as.data.frame(RANN::nn2(df2[,c(2,3)],df1[,c(2,3)],k=1))

#    nn.idx   nn.dists
# 1       5  0.7990743
# 2       8  2.1676868
# 3      10  1.4758202
# 4       3  4.2415854
# 5       2  3.6445416
# 6       3  2.3577811
# 7       9  8.2123762
# 8       7 11.4936496
# 9       1  1.9296370
# 10      2  3.2180026
...