向量化思维

Question

我有向量化问题。假设我有一个向量x <- c(0,0,1,0,1,1,0)，我想要的是得到向量(1,0,1,1)，或者把0从第一个点移到第一个非零点，所有0从最后一个非零点移到最后一个点。这在正常情况下很容易，但棘手的部分是我不能使用任何循环(for、while等)。基本上，我必须“矢量化”整个算法。有人帮忙吗？

Answer 1

我想你可以试试这个，谢谢@JacobH的评论，这样会更快。

x <- c(0,0,1,0,1,1,0)
zeros <- which(x > 0)

x[zeros[1]:tail(zeros, n = 1)]

输出量

1 1 0 1 1

Answer 2

好吧，如果速度是个问题，只需使用循环：

x <- rpois(1e8,1)
# Zihu Guo & JacobH's answer
system.time({zeros <- which(x > 0); x[zeros[1]:tail(zeros, n = 1)]})
#    user  system elapsed 
#    1.46    0.42    1.89 

# a slightly improved version of their answer
system.time({zeros <- which(!!x); x[zeros[1]:tail(zeros, n = 1)]})
#    user  system elapsed 
#    1.21    0.42    1.64 


system.time({
    lo  = 0L
    lov = 0L
    while (!lov){ lo = lo+1L ; lov = lov+x[lo] }
    hi  = length(x)+1L
    hiv = 0L
    while (!hiv){ hi = hi-1L ; hiv = hiv+x[hi] }    
    x[lo:hi]
})
#    user  system elapsed 
#    0.62    0.07    0.71 

除了速度，这是很好的，因为它不需要真正的知识R，它可以使它更快的Rcpp包，可能。

(如果整个向量为零，则永远不会终止。如果这件事重要的话，可以做一些小的调整。如果向量大部分为零，则which方法速度更快。例如，以x <- rep(0L,1e8); x[1e8/2] <- 1L为例。