R：使用sapply加快循环速度

Question

我是R语言的新手，我想弄清楚如何使用sapply来提高我的功能，有帮助吗？

x <- c(0.0, 1.0, 2.0, 3.0, 4.0, 5.7)
y <- c(10.0, 7.0, 5.5, 4.5, 3.2, NA)
z <- c(as.data.frame(cbind(x,y))


vcub=function(y, x, z) 
{
    vol<-vector()
    for(i in 1:dim(z)[1]){
        if(is.na(y[i]))
        {
          vol[i]<-(((pi*y[i-1]^2)/40000)/3)*(x[i]-x[i-1])
        }else{
          vol[i]<-(((pi*y[i-1]^2) + (pi*y[i]^2))/80000)*(x[i]-x[i-1])
        }
    }
    return(as.data.frame(vol))
}    

Answer 1

您可以通过将if循环中的if和else语句替换为ifelse语句，并在R中使用矢量化算术来矢量化此代码：

data.frame(vol=ifelse(is.na(y), pi*c(NA, head(y, -1)^2)/120000*c(NA, diff(x)),
                      pi*(y^2 + c(NA, head(y, -1)^2))/80000*c(NA, diff(x))))
#            vol
# 1           NA
# 2 0.0058512163
# 3 0.0031121402
# 4 0.0019831304
# 5 0.0011973395
# 6 0.0004557404

通常，当您可以在不使用任何先前计算的索引的情况下计算结果的第i个索引时，很容易对计算进行向量化。 由于您的vol[i]公式不依赖于任何先前的vol值，因此您可以使用基本算术运算符。