Question

当然是一个facepalm问题，抱歉。我一直在尝试RTFM并谷歌，但没有运气。

我试图在R中创建一个愚蠢的函数，它将一个向量作为参数，使用for循环遍历它，并对每个组件进行非常简单的添加。

func1 <- function(vector) {
  vr <- c() ### an empty vector 

  for (i in 1:length(vector)) { 
    vr <- vector[i]+i
  }
}

对我来说，这看起来是正确的，但我总是得到一个NULL向量。为什么？

Answer 1

你可以像这样写得更有效率：

func1 <- function(vector){
    vector + seq_along(vector)
}

这样你就可以利用矢量加法了。

为了完整起见，让我们对这两种解决方案进行基准测试。这是一般循环解决方案：

func2 <- function(vector) {
    vr <- c() ### an empty vector 

    for (i in 1:length(vector)) { 
        vr[i] <- vector[i]+i
    }
    vr
}

作为一个升级，还有就地解决方案

func3 <- function(vector) {    
    for (i in 1:length(vector)) { 
        vector[i] <- vector[i]+i
    }
    vector
}

使用microbenchmark我们可以看到向量化解决方案是迄今为止效率最高的。

vec <- sample(100000, 10000)

library(microbenchmark)
microbenchmark(func1(vec), func3(vec), func2(vec))

Unit: microseconds
       expr       min        lq         mean    median       uq         max neval cld
 func1(vec)    29.998    36.984     44.78312    42.736    44.38     399.006   100  a 
 func3(vec) 12845.823 13666.432  14452.02863 14060.712 14708.53   25025.950   100  a 
 func2(vec) 84898.055 87354.750 110046.26659 88634.566 91193.38 1042819.269   100   b

Answer 2

您的代码很接近但有两个问题。首先，除非使用return()调用指定，否则R中的函数将返回最后一个表达式。在你的函数中，这是for循环本身。但for循环表达式本身并不返回任何内容。

其次，最后一行不会在vr上编制索引，因此您将使用循环中的最后一个结果完全覆盖它。

因此，这段代码（我认为）会做你想要的。但同意其他答案，有更简单的方法来做到这一点。

func1 <- function(vector) {
  vr <- c() ### an empty vector 

  for (i in 1:length(vector)) { 
    vr[i] <- vector[i]+i
  }
  vr
}

这给出了这个输出：

> func1(c(1,2))
[1] 2 4

R函数在向量上使用FOR循环

2 个答案: