R 使用 if 下次中断 for 循环时出错

Question

我正在尝试在for循环中使用if语句，以便如果不满足if条件，R 将停止运行该i值的循环，但将继续运行整个循环，我认为我遇到了一个(简单？(语法错误，它给了我与预期不同的结果。

下面是我的示例数据集和我的循环：

test <- matrix(ncol=3, nrow=3, data = c(10, 20, 30, NA, 10, 20, 10, 30, 20), byrow = T)
mean_fun<-function(x,y){
mean <- sum(x,y)/2
return(mean)
}
for (i in (1:nrow(test))){
var1 <- vector()
var2 <- vector()
var1[i] <- mean_fun(test[i,1], test[i,2])
if (is.na(var1[i]==TRUE)) next
var2[i] <- mean_fun(var1[i], test[i,3])
print(var2)
}

如果我只是计算循环中的var1值，var1 = c(15, NA, 20).如果我的var1 == NA，我不想计算var2.简而言之，我想从这个循环中获得的结果是var2 = c(22.5, NA, 20.0).

但是当我运行这个循环时，我得到了[1] 22.5 [1] NA NA 20这意味着循环运行正常i=1，也许i=2但在i=2或i=3期间搞砸了？我无法弄清楚语法是如何工作的，因此它给了我NA NA 20.任何建议将不胜感激！

Answer 1

您需要在循环之前启动向量，而不是每次迭代都在循环内启动。此外，在循环后检查结果，而不是每次迭代。

var1 <- vector()
var2 <- vector()
for (i in (1:nrow(test))){
var1[i] <- mean_fun(test[i,1], test[i,2])
if (is.na(var1[i]==TRUE)) next
var2[i] <- mean_fun(var1[i], test[i,3])
}
var1
# [1] 15 NA 20
var2
# [1] 22.5   NA 20.0