R *应用向量作为输入;矩阵作为输出

时间:2015-11-09 10:41:50

标签: r matrix vector apply

我想在矢量的每个元素上应用一个输出矢量的函数。 在将该函数应用于该向量的每个元素之后,我应该有许多向量,我希望rbind为了得到一个矩阵。

代码应该等同于以下内容:

my_function <- function(x) x:(x+10)

my_vec <- 1:10
x <- vector()
for(i in seq_along(vec)){
  x <- rbind(x,my_function(my_vec[i]))
}

当然,my_functionmy_vec只是示例。

2 个答案:

答案 0 :(得分:2)

尝试:

 tmp <- lapply(my_vec, my_function)
 do.call(rbind, tmp)

或者像Heroka建议的那样,使用sapply。我更喜欢lapply,然后按我喜欢的方式绑定我的输出(rbind / cbind),而不是潜在的转置。

答案 1 :(得分:1)

这是另一种选择:

matrix( unlist(lapply(my_vec,my_function)), length(my_vec), byrow=TRUE )

速度几乎相同:

library(microbenchmark)

my_function <- function(x) sin(x:(x+10))

for ( n in 1:4 )
{
  my_vec <- 1:10^n

  print(
    microbenchmark( mra68 = matrix( unlist(lapply(my_vec,my_function)), length(my_vec), byrow=TRUE ),
                    stas.g = do.call(rbind, lapply(my_vec, my_function)),
                    times = 1000 )
  )

  print("identical?")
  print( identical( matrix( unlist(lapply(my_vec,my_function)), length(my_vec), byrow=TRUE ),
                    do.call(rbind, lapply(my_vec, my_function)) ) )  
}

Unit: microseconds
   expr    min     lq     mean median      uq     max neval
  mra68 38.496 40.307 68.00539 41.213 110.052 282.148  1000
 stas.g 41.213 42.572 72.86443 43.930 115.939 445.186  1000
[1] "identical?"
[1] TRUE
Unit: microseconds
   expr     min      lq     mean   median       uq      max neval
  mra68 793.002 810.212 850.4857 818.3640 865.2375 7231.669  1000
 stas.g 876.786 894.901 946.8165 906.2235 966.9100 7051.873  1000
[1] "identical?"
[1] TRUE
Unit: milliseconds
   expr      min       lq     mean   median       uq      max neval
  mra68 2.605448 3.028442 5.269003 4.020940 7.807512 14.51225  1000
 stas.g 2.959604 3.390071 5.823661 4.500546 8.800462 92.54977  1000
[1] "identical?"
[1] TRUE
Unit: milliseconds
   expr      min       lq     mean   median       uq      max neval
  mra68 27.29810 30.99387 51.44223 41.20167 79.46185 559.0059  1000
 stas.g 33.63622 37.22420 60.10224 49.07643 92.94333 395.3315  1000
[1] "identical?"
[1] TRUE
>