返回基于一个data.frame中的值的向量和另一个

时间:2015-11-05 19:34:48

标签: r loops apply lapply mapply

我有一个data.frame A,包含+100 000行,包含位置信息(Lat =纬度,Lon =经度),另一个data.frame B包含+500行,包含地理区域限制(区域不重叠)。

如何根据A中的位置创建一个从B返回地名的函数?

我希望R在B中存在位置时返回地名矢量,在不存在位置的情况下返回“未分配”,在A中缺少Lat或Lon的情况下返回NA

示例:

dataA <- structure(list(Lat = c(60L, 63L, 0L, 75L, NA, 71L, 70L), Lon = c(-70L, 
-66L, 5L, -100L, 80L, -61L, -150L)), .Names = c("Lat", "Lon"), class = "data.frame", row.names = c(NA, 
-7L))

dataB <- structure(list(Region = structure(c(2L, 3L, 1L), .Label = c("Beaufort Sea", 
"Hudson Strait", "North West Passage"), class = "factor"), Lat.min = c(55, 
70, 69.5), Lat.max = c(65L, 80L, 72L), Lon.min = c(-75L, -120L, 
-160L), Lon.max = c(-60L, -60L, -120L)), .Names = c("Region", 
"Lat.min", "Lat.max", "Lon.min", "Lon.max"), class = "data.frame", row.names = c(NA, 
-3L))

## I would like to test for each row in dataA:

i <- 1 ## i <- 1:nrow(dataB)
dataA$Lat > dataB$Lat.min[i] & dataA$Lat < dataB$Lat.max[i] &
dataA$Lon > dataB$Lon.min[i] & dataA$Lon < dataB$Lon.max[i]

## and return                                                                                                              
dataB$Region[i]  ## only once for each row of dataA,                                                                          
##unless is.na(dataA$Lat) | is.na(dataA$Lon), then return(NA),
##and if a row in dataA does not match any row in dataB, then return "not assigned"

## The result should look something like:
c("Hudson Strait", "Hudson Strait", "not assigned", "North West Passage", 
NA, "North West Passage", "Beaufort Sea")

我尝试了什么:我之前使用ifelse函数解决了类似的挑战,但在这种情况下,我的条件data.frame太大而无法手动执行此操作。我还尝试split data.frame A并运行条件for循环,但没有设法弄清楚如何制定for循环。如果我在循环中使用if语句,我的循环返回的值与A中每行的B值中的行一样多。我还有一种感觉,即为此数据集运行for循环需要很长时间时间,更不用说拆分数据的大小。框架A.必须有更好的方法来做到这一点......

1 个答案:

答案 0 :(得分:0)

这是一种,当然不是最优雅的方式:

z <- lapply(1:nrow(dataB), function(i){
  ifelse(is.na(dataA$Lat) | is.na(dataA$Lon), "Missing", 
ifelse(dataA$Lat > dataB$Lat.min[i] & dataA$Lat < dataB$Lat.max[i] &
 dataA$Lon > dataB$Lon.min[i] & dataA$Lon < dataB$Lon.max[i], 
as.character(dataB$Region[i]), NA))
  })

z <- do.call(rbind,z)

apply(z, 2, function(j) {
  out <-j[!is.na(j)]
  if(length(out) == nrow(z)) {
    return(NA)} else {
      if(length(out) > 0) {
      return(out)} else {
        return("Not assigned")
        }}
  })

# [1] "Hudson Strait" "Hudson Strait" "Not assigned" "North West Passage" 
# NA "North West Passage" "Beaufort Sea" 

也许某人有更优雅的解决方案?