代码之家  ›  专栏  ›  技术社区  ›  Neal Barsch

按组计算跨越数百万坐标的最大距离

  •  3
  • Neal Barsch  · 技术社区  · 6 年前

    在R中按组计算一组坐标之间的最大距离最有效的方法是什么?

    样本数据: 我有这样的数据,但我有超过2500万条条目,而不是x10000(例如)。

    library(data.table)
    data <- data.table(latitude=sample(seq(0,90,by=0.001), 10000, replace = TRUE),
                   longitude=sample(seq(0,180,by=0.001), 10000, replace = TRUE))
    groupn <- nrow(data)/1000
    data$group <- sample(seq(1,groupn,by=1),10000,replace=T)
    

    我现在的方法很慢:

    data <- data[order(data$group),]
    library(dplyr)
    library(sf)
    library(foreach)
    distlist <- foreach(i=1:10)%do%{
      tempsf <- st_as_sf(filter(data,group==i), coords= c("longitude", "latitude"), crs=4326)
      max(st_distance(tempsf, tempsf))
      }
    

    有什么天才能帮我加快速度吗?

    2 回复  |  直到 6 年前
        1
  •  2
  •   Juan Antonio Roldán Díaz    6 年前

    试试这个:

    欧几里得扩张:

    > system.time(out1 <- tapply(1:nrow(data), data$group, function(x) max(dist(data[x, 1:2]))))
       user  system elapsed 
       0.14    0.00    0.14 
    > out1
       1        2        3        4        5        6        7        8        9       10 
    199.2716 197.1172 194.7018 197.2652 196.3747 197.6728 194.7344 197.8781 195.3837 195.0123 
    

    WGS84:

    > auxF <- function(x) {
    +   require(sp)
    +   
    +   tempsf <- data[x, 1:2]
    +   coordinates(tempsf) <- c("longitude", "latitude")
    +   proj4string(tempsf) = "+proj=longlat +ellps=WGS84 +no_defs"
    +   return(max(spDists(tempsf)))
    + }
    > 
    > system.time(out2 <- tapply(1:nrow(data), data$group, auxF))
       user  system elapsed 
       4.71    0.00    4.76 
    > out2
       1        2        3        4        5        6        7        8        9       10 
    19646.04 19217.48 19223.27 19543.99 19318.55 18856.65 19334.11 19679.45 18840.90 19460.14 
    

    Haversine方法:

    > system.time(out3 <- tapply(1:nrow(data), data$group, function(x) max(distm(as.matrix(data[x,.(longitude,latitude)], fun=distHaversine)))))
       user  system elapsed 
      13.24    0.01   13.30 
    > out3
       1        2        3        4        5        6        7        8        9       10 
    19644749 19216989 19223012 19542956 19317958 18856273 19333424 19677917 18840641 19459353 
    

    对于700万条记录,你可以假设一个欧几里得距离或者把你的点投射到一个平面上,这样你就可以使用欧几里得距离,因为我们知道最大的距离是在每个组的凸壳的点之间,这大大减少了操作,而且它不需要一个L。RAM的OT:

    > system.time(out4 <- tapply(1:nrow(data), data$group, function(x) max(dist(data[x, 1:2][chull(data[x, 1:2]), ]))))
       user  system elapsed 
       0.03    0.00    0.03 
    > out4
           1        2        3        4        5        6        7        8        9       10 
    199.2716 197.1172 194.7018 197.2652 196.3747 197.6728 194.7344 197.8781 195.3837 195.0123 
    

    大数据:

    > data <- data.table(latitude=sample(seq(0,90,by=0.001), 7000000, replace = TRUE),
    +                    longitude=sample(seq(0,180,by=0.001), 7000000, replace = TRUE))
    > groupn <- nrow(data)/700000
    > data$group <- sample(seq(1,groupn,by=1),7000000,replace=T)
    > 
    > system.time(out1 <- tapply(1:nrow(data), data$group, function(x) max(dist(data[x, 1:2]))))
    Error: cannot allocate vector of size 1824.9 Gb
    Called from: dist(data[x, 1:2])
    Browse[1]> 
    Timing stopped at: 7.81 0.06 7.91
    > system.time(out4 <- tapply(1:nrow(data), data$group, function(x) max(dist(data[x, 1:2][chull(data[x, 1:2]), ]))))
       user  system elapsed 
       8.41    0.22    8.64 
    
        2
  •  2
  •   Neal Barsch    6 年前

    多亏了胡安·安东尼奥,他想用塔普里。…最后我把这个函数用到了你建的SP中,它是最快的。

    auxF <- function(x) {
    require(sp)
    tempsf <- data[x, 1:2]
    coordinates(tempsf) <- c("longitude", "latitude")
    proj4string(tempsf) = "+proj=longlat +ellps=WGS84 +no_defs"
    return(max(spDists(tempsf)))
    }
    out1 <- tapply(1:nrow(data), data$group, auxF)
    

    这也适用于: 哈弗斯先生@ SymbolixAU (像往常一样棒极了) built :

    dt.haversine <- function(lat_from, lon_from, lat_to, lon_to, r = 6378137){
      radians <- pi/180
      lat_to <- lat_to * radians
      lat_from <- lat_from * radians
      lon_to <- lon_to * radians
      lon_from <- lon_from * radians
      dLat <- (lat_to - lat_from)
      dLon <- (lon_to - lon_from)
      a <- (sin(dLat/2)^2) + (cos(lat_from) * cos(lat_to)) * (sin(dLon/2)^2)
      return(2 * atan2(sqrt(a), sqrt(1 - a)) * r)
    }
    library(geosphere)
    out1 <- tapply(1:nrow(data), data$group, function(x) max(distm(as.matrix(data[x,c("longitude","latitude")], fun=dt.haversine))))