在循环中同时使用子集和整个数据帧

Question

我正在尝试编写一个循环遍历数据帧行并使用有关其他行的信息来确定每个循环的输出的函数。

考虑以下数据框，它表示具有经度坐标、纬度坐标和表示他们是否生病的值的人：

game.mat<-as.data.frame(matrix(0, nrow = 100, ncol = 3))
colnames(game.mat)<-c("PosX","PosY","Sick")
game.mat[,"PosX"]<-sample(x = c(1:100), 100, replace = TRUE)
game.mat[,"PosY"]<-sample(x = c(1:100), 100, replace = TRUE)
game.mat[,"Sick"]<-sample((c(rep(0,8),rep(1,2))),100,replace=TRUE)

在基线时，少数人会生病。我的功能是感染与病人有相邻 x-y 坐标的人（因此任何与病人相邻的人）。我考虑在 ifelse 语句中嵌入这样的函数：

 search_sick<-function(d,corx,cory){
   d2<-d[d$PosX<corx+2&d$PosX>corx-2&d$PosY<cory+2&d$PosY>cory-2,]
   if(sum(d2$Sick>0)){
    d$Sick<-1
   } else{
    d$Sick<-0
   }
  } 

但它会让每个人都生病，也许是因为如果有人在病人旁边，它会给每个人一个 1 的值。我还考虑过使用应用功能。但是根据我对 apply 的了解，它一次只能在一行内运行，因此无法检索有关其他行是否具有相邻坐标值的信息。

我希望这是有道理的。很高兴提供任何其他信息。

Answer 1

这是一个使用apply的例子

set.seed(1)

game.mat<-as.data.frame(matrix(0, nrow = 100, ncol = 3))
colnames(game.mat)<-c("PosX","PosY","Sick")
game.mat[,"PosX"]<-sample(x = c(1:100), 100, replace = TRUE)
game.mat[,"PosY"]<-sample(x = c(1:100), 100, replace = TRUE)
game.mat[,"Sick"]<-sample((c(rep(0,8),rep(1,2))),100,replace=TRUE)

#plot the sick individuals in red
plot(PosY~PosX, data=game.mat, col=as.factor(Sick), pch=16) 

我们将修改您的函数，使其具有灵活的搜索半径“r”，并返回新感染个体的索引

search_sick<-function(d, corx, cory, r){
  indx<-which(d$PosX<corx+r & d$PosX>corx-r & d$PosY<cory+r & d$PosY>cory-r)
}

contagious<-game.mat[game.mat$Sick==1,]

infected<-apply(contagious, 1, function(x) {
  search_sick(game.mat, x[1], x[2], r=10)
})

game.mat$T1<-game.mat$Sick

game.mat$T1[unique(unlist(infected))]<-1

#circle points which have become sick
points(PosY~PosX, data=game.mat[game.mat$Sick==0 & game.mat$T1==1,], col="red", cex=2)