R 使用 apply 避免 for 循环

Question

我有这样的问题。我使用这种语句来管理它，但速度很慢。我可以使用应用函数提高性能吗？

x<-matrix(sample(1:100,40),20,2)
x<-as.data.frame(x)
for ( i in 1:nrow(x))
{
  if ( x[i,1]>x[i,2] ){
    x[i,3]<-1
  } else {
    x[i,3]<-0
  }
}

Answer 1

您不需要显式或隐式循环；只需使用 ifelse:

x[, 3] <- ifelse(x[, 1] > x[, 2], 1, 0)

事实上，您可以进一步简化：

x[, 3] <- x[, 1] > x[, 2]

这将创建一个逻辑列而不是数字列，但对于大多数用途来说已经足够接近了。如果不是，您可以将逻辑转换回数字：

x[, 3] <- as.numeric(x[, 1] > x[, 2])

R avoid a for loop using an apply