我是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 是包含现有圆圈的临时数据框。
你可以看到这些圆圈用于什么,上面的红色圆圈有21个其他位置适合它的半径(21 + 1原始= 22)
非常感谢您的帮助, 阿伦娜
答案 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
行,其中n
是circles
中的行数数据框。从这里开始,您可以进行进一步处理,例如为多个圆圈中的每个点保留一行等等。
这是一个例子。我们将返回一个数据框列表,其中一个点位于其中,或者返回&#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))
你最终会得到不同的坐标想法。但是,您的数据顺序会丢失。