Question

我有一堆向量需要将一个向量相加。我正在寻找更优雅的矢量添加解决方案，而不是使用'+'运算符。有没有人知道以更舒适的方式做这件事的任何伎俩。感谢

载体：

a <- c(1,1,0,2,1,0,1,0,1)
b <- c(0,0,1,0,1,1,0,1,0)
c <- c(0,1,1,0,0,2,1,1,1)

我知道做这个的虚拟方式，我期待这样做的优雅

期望的输出：

out <- c(1,2,2,2,2,3,2,2,2)

更有效地进行此类操作的优雅方式是什么？

Answer 1

我们可以使用rbind将所有向量放在一起，然后使用colSums：

colSums(rbind(a, b, c))
# [1] 1 2 2 2 2 3 2 2 2

基准：

# bigger input
set.seed(1)
n <- 10^7
a <- runif(n)
b <- runif(n)
c <- runif(n)
d <- runif(n)
e <- runif(n)
f <- runif(n)

# benchmark
microbenchmark::microbenchmark(
  colSums = colSums(rbind(a, b, c, d, e, f)),
  rowSums = rowSums(cbind(a, b, c, d, e, f)),
  Reduce = base::Reduce("+", list(a, b, c, d, e, f)),
  S4vReduce = S4Vectors::Reduce('+', lapply(list(a, b, c, d, e, f), lengths)),
  JustAdd = a + b + c + d + e + f
)


# Unit: milliseconds
#       expr       min        lq      mean    median       uq      max neval   cld
#    colSums 408.31052 427.94015 470.27181 461.18763 494.1420 651.3383   100     e
#    rowSums 349.93752 359.15854 408.82652 397.99315 434.1662 569.3575   100    d 
#     Reduce 129.43443 134.55584 183.34432 179.88746 208.0281 339.9345   100  b   
#  S4vReduce 162.90015 166.19150 206.16387 192.73739 212.2146 380.2038   100   c  
#    JustAdd  73.38243  74.00267  92.68309  76.12524  82.7517 282.6101   100 a

Answer 2

使用来自S4Vectors的Reduce：

vec.li <- list(a,b,c)
vec.sum <- S4Vectors::Reduce('+', lapply(vec.li, lengths))

这个解决方案可以适用于添加非常大的维度向量，到目前为止快速有效的情况。

如何更优雅地做矢量和而不是使用'+'运算符

2 个答案: