R - 根据位置

时间:2016-11-30 20:46:16

标签: r optimization

我是R的新人,但我听说使用for循环真是个坏主意。我有使用它们的工作代码,但我想改进它,因为它对于大数据来说非常慢。我已经有了一些如何改进算法的想法,但我不知道如何对其进行矢量化,或者在没有for循环的情况下进行。

我只是将lat / lng点分组为半径为参数的圆圈。

该函数的示例输出(仅填充circle_id列中的值),radius设置为100米:

[1] "Locations: "
   latitude  longitude sensor_time sensor_time2         circle_id
   48.15144  17.07569  1447149703  2015-11-10 11:01:43         1
   48.15404  17.07452  1447149743  2015-11-10 11:02:23         2
   48.15277  17.07514  1447149762  2015-11-10 11:02:42         3
   48.15208  17.07538  1447149771  2015-11-10 11:02:51         1
   48.15461  17.07560  1447149773  2015-11-10 11:02:53         4
   48.15139  17.07562  1447149811  2015-11-10 11:03:31         1
   48.15446  17.07517  1447149866  2015-11-10 11:04:26         2
   48.15266  17.07330  1447149993  2015-11-10 11:06:33         5

所以我有2个for循环,loop1遍历每一行,loop2遍历每个前面的circle_id,并检查loop1的当前位置是否在loop2的现有圆的半径范围内。每个circle_id的中心是在前一个半径之外找到的第一个位置。

以下是代码:

init_circles = function(datfr, radius) {
  cnt = 1
  datfr$circle_id[1] = 1
  longitude = datfr$longitude[1]
  latitude = datfr$latitude[1]
  circle_id = datfr$circle_id[1]
  datfr2 <- data.frame(longitude, latitude, circle_id)

  for (i in 2:NROW(datfr)) {
      for (j in 1:NROW(datfr2)) {
        tmp = distHaversine(c(datfr$longitude[i],datfr$latitude[i]) ,c(datfr2$longitude[j],datfr2$latitude[j]))
        if (tmp < radius){
          datfr$circle_id[i] = datfr2$circle_id[j]
          break
        }
      }
      if (datfr$circle_id[i]<1){
        cnt = cnt +1
        datfr$circle_id[i] = cnt
        datfr2[nrow(datfr2)+1,] = c(datfr$longitude[i],datfr$latitude[i],datfr$circle_id[i])
      }
  }
  return(datfr)
}

datfr 是没有设置circle_id的输入数据框, datfr2 是包含现有圆圈的临时数据框。

编辑:这是一个视觉输出:
enter image description here

你可以看到这些圆圈用于什么,上面的红色圆圈有21个其他位置适合它的半径(21 + 1原始= 22)

非常感谢您的帮助, 阿伦娜

2 个答案:

答案 0 :(得分:2)

我假设我们有一个数据框circles,其中包含每个圆圈的中心和半径,并且您问题中发布的样本数据位于名为dat的数据框中。下面的代码矢量化距离计算,并使用lapply计算每个点距每个圆心的距离,并确定每个点是否在该圆的半径范围内。

library(geosphere)

# We'll check the distance of each data point from the center of each 
#  of these circles
circles = data.frame(ID=1:2, lon=c(17.074, 17.076), lat=c(48.1513, 48.15142), 
                     radius=c(180,190))

datNew = lapply(1:nrow(circles), function(i) {

  df = dat

  df$dist = distHaversine(df[,c("longitude", "latitude")], 
                          circles[rep(i,nrow(df)), c('lon','lat')])

  df$in_circle = ifelse(df$dist <= circles[i, "radius"], "Yes", "No")

  df$circle_id = circles[i, "ID"]

  df

})

datNew = do.call(rbind, datNew)

datNew
   latitude longitude sensor_time sensor_time2    time3      dist in_circle circle_id
1  48.15144  17.07569  1447149703   2015-11-10 11:01:43 126.47756       Yes         1
2  48.15404  17.07452  1447149743   2015-11-10 11:02:23 307.45048        No         1
3  48.15277  17.07514  1447149762   2015-11-10 11:02:42 184.24465        No         1
4  48.15208  17.07538  1447149771   2015-11-10 11:02:51 134.32601       Yes         1
5  48.15461  17.07560  1447149773   2015-11-10 11:02:53 387.15358        No         1
6  48.15139  17.07562  1447149811   2015-11-10 11:03:31 120.73138       Yes         1
7  48.15446  17.07517  1447149866   2015-11-10 11:04:26 362.34236        No         1
8  48.15266  17.07330  1447149993   2015-11-10 11:06:33 160.07179       Yes         1
9  48.15144  17.07569  1447149703   2015-11-10 11:01:43  23.13059       Yes         2
10 48.15404  17.07452  1447149743   2015-11-10 11:02:23 311.68096        No         2
11 48.15277  17.07514  1447149762   2015-11-10 11:02:42 163.29068       Yes         2
12 48.15208  17.07538  1447149771   2015-11-10 11:02:51  86.70762       Yes         2
13 48.15461  17.07560  1447149773   2015-11-10 11:02:53 356.34955        No         2
14 48.15139  17.07562  1447149811   2015-11-10 11:03:31  28.41890       Yes         2
15 48.15446  17.07517  1447149866   2015-11-10 11:04:26 343.97933        No         2
16 48.15266  17.07330  1447149993   2015-11-10 11:06:33 243.44024        No         2

所以我们现在有一个数据框告诉我们每个点是否在给定的圆内。数据框采用长格式,这意味着原始数据框n中的每个点都有dat行,其中ncircles中的行数数据框。从这里开始,您可以进行进一步处理,例如为多个圆圈中的每个点保留一行等等。

这是一个例子。我们将返回一个数据框列表,其中一个点位于其中,或者返回&#34;无&#34;如果该点不在任何圆圈内:

library(dplyr)

datNew %>%
  group_by(latitude, longitude) %>% 
  summarise(in_which_circles = if(any(in_circle=="Yes")) paste(circle_id[in_circle=="Yes"], collapse=",") else "None")
  latitude longitude in_which_circles
     <dbl>     <dbl>            <chr>
1 48.15139  17.07562              1,2
2 48.15144  17.07569              1,2
3 48.15208  17.07538              1,2
4 48.15266  17.07330                1
5 48.15277  17.07514                2
6 48.15404  17.07452             None
7 48.15446  17.07517             None
8 48.15461  17.07560             None

答案 1 :(得分:0)

在我看来,使用 for 循环并不是一个坏主意,有时我更喜欢使用循环来使我的代码更清晰,而不是嵌套 apply 。< / p>

但是在你的情况下,你可以尝试这样的事情:

library(dplyr)
library(tidyr)
library(purrr)

# I only load the coordinate for now
df <- tibble(latitude  = c(48.15144, 48.15404, 48.15277, 48.15208, 48.15461, 48.15139, 48.15446, 48.15266),
             longitude = c(17.07569, 17.07452, 17.07514, 17.07538, 17.07560, 17.07562, 17.07517, 17.07330))

df %>% 
  unite(coord, latitude, longitude, sep = ", ") %>% 
  mutate(coord2 = coord) %>% 
  expand(coord, coord2) %>% 
  mutate(coord = map(coord, function(x) {xx <- as.numeric(unlist(strsplit(x, ","))); tibble(lat = xx[1], lon = xx[2])})) %>% 
  mutate(coord2 = map(coord2, function(x) {xx <- as.numeric(unlist(strsplit(x, ","))); tibble(lat2 = xx[1], lon2 = xx[2])})) %>% 
  unnest() %>% 
  rowwise() %>% 
  mutate(dist = distHaversine(c(lon, lat), c(lon2, lat2))) %>% 
  group_by(lat, lon) %>% 
  mutate(group = 1:n()) %>% 
  group_by(group) %>% 
  filter(dist < 100) %>% 
  group_by(lat, lon) %>% 
  summarise(group = min(group))

你最终会得到不同的坐标想法。但是,您的数据顺序会丢失。