r - 如何填充向量中的值？

Question

我在 R 中有包含很多 0 和一些非零数字的向量。每个向量都以非零数字开头。

例如 <1,0,0,0,0,0,2,0,0,0,0,0,4,0,0,0>

我想将所有零设置为等于最近的非零数字。

IE。这个向量将变成 <1,1,1,1,1,1,2,2,2,2,2,2,4,4,4,4>

我需要为大约 100 个向量执行此操作，每个向量包含大约 600 万个条目。目前我正在使用 for 循环:

for(k in 1:length(vector){

  if(vector[k] == 0){

    vector[k] <- vector[k-1]
  }
}

有没有更有效的方法来做到这一点？

谢谢!

Answer 1

一种选择是用 0 替换那些 NA ，然后使用 zoo::na.locf :

x <- c(1,0,0,0,0,0,2,0,0,0,0,0,4,0,0,0)
x[x == 0] <- NA
zoo::na.locf(x)  ## you possibly need: `install.packages("zoo")`
# [1] 1 1 1 1 1 1 2 2 2 2 2 2 4 4 4 4

感谢 Richard 向我展示了如何使用 replace ，

zoo::na.locf(replace(x, x == 0, NA))