R fill vector with same value until a different one is found

Question

In R I have two vectors such as

2 NA NA  1 NA NA
1 NA  2 NA NA NA

And I would like them to be like

2 2 2 1 1 1
1 1 2 2 2 2

Any ideas?

See `zoo::na.locf` from the `zoo` package. – nicola Nov 15 '18 at 14:21 — nicola, Nov 15 '18 at 14:21

score 0 · Answer 1 · answered Nov 15 '18 at 14:29

With tidyverse:

 read.table(text="2 NA NA  1 NA NA
                  1 NA  2 NA NA NA")%>%
    t()%>%
   data.frame()%>%
   mutate_all(funs(as.numeric(.)))%>%
   fill(c(X1,X2))%>%t

   [,1] [,2] [,3] [,4] [,5] [,6]
X1    2    2    2    1    1    1
X2    1    1    2    2    2    2

score 0 · Answer 2 · answered Nov 15 '18 at 14:35

vectors:

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

As anonymous function:

(function(x) {
    nai <- which(is.na(x))
    i <- which(!is.na(x))

    x[nai] <- x[i][findInterval(nai,i)]
    return(x)
 })

Call:

    (function(x) {
        nai <- which(is.na(x))
        i <- which(!is.na(x))

        x[nai] <- x[i][findInterval(nai,i)]
        return(x)
     })(a)
#[1] 2 2 2 1 1 1

    (function(x) {
        nai <- which(is.na(x))
        i <- which(!is.na(x))

        x[nai] <- x[i][findInterval(nai,i)]
        return(x)
     })(b)
#[1] 1 1 2 2 2 2

R fill vector with same value until a different one is found

2 Answers2