r - 使用条件循环遍历 data.table 行

Question

我有一个包含 ID 和位置的 data.table。例如，这里有一行:(它有 col 和 row 名称，不知道是否重要)

locations<-data.table(c(11,12),c(-159.58,0.2),c(21.901,22.221))
colnames(locations)<-c("id","location_lon","location_lat")
rownames(locations)<-c("1","2")

然后我想遍历这些行并将它们与另一个点(使用纬度、经度)进行比较。在 for 循环中它起作用:

for (i in 1:nrow(locations)) {
    loc <- locations[i,]
    dist <- gdist(-159.5801, 21.901, loc$location_lon, loc$location_lat, units="m")
    if(dist <= 50) {
        return (loc)
    }
    return (NULL)
}

并返回:

id location_lon location_lat

1: 11 -159.58 21.901

但我想使用申请。以下代码无法运行:

dists <- apply(locations,1,function(x) if (50 - gdist(-159.5801, 21.901, x$location_lon, x$location_lat, units="m")>=0) x else NULL)

$ operator is invalid for atomic vectors 错误。更改为按位置引用 (x[2],x[3]) 不足以解决此问题，我明白了

Error in if (radius - gdist(lon, lat, x[2], x[3], units = "m") >= 0) x else NULL : 
missing value where TRUE/FALSE needed 

这是因为data.table被转成了矩阵，坐标被当成了文本而不是数字。有没有办法克服这个问题？该解决方案需要高效(我想对 >1,000,000 个不同的坐标运行此检查)。如果需要，可以更改位置表的数据结构。

Answer 1

不需要循环，只需按预期使用data.table。如果您只想查看距所需位置 50 米范围内的行，则只需

locations[, if (gdist(-159.58, 21.901, location_lon, location_lat, units="m") <= 50) .SD, id]
##    id location_lon location_lat
## 1: 11      -159.58       21.901

在这里，我们通过 locations 数据集本身中的 id 列进行迭代，并检查每个 id 是否在距离 50 米以内>-159.58, 21.901。如果是这样，我们将调用 .SD，它基本上是特定 id 的数据集本身。

---

作为旁注，data.table 没有row.names，因此无需指定它们，参见here , 例如