从上一个非NA值获取索引

Question

我有这个载体

set.seed(234)
x <- sample(c(rep(NA,3),1:5))
x
[1]  3  5 NA  1  4 NA NA  2

对于每个NA ，我想要最后一个非NA值的索引（或值）。 也就是说，对于第一个NA ，最后一个非NA具有索引2.对于接下来的两个NA ，它们的最后一个非NA具有索引5：

[1]  NA  NA 2  NA  NA 5 5  NA

Base R或tidyverse就可以了。 我尝试了lag ， lead ， rle ， gl和coalesce ，但都没有成功。 例如，这非常接近，但仍然是错误的。

a <- rle(is.na(x))
rep(1:length(a$lengths), a$lengths)
[1] 1 1 2 3 3 4 4 5

Answer 1

使用基数R.

y=1:length(x)
y[is.na(x)]=0
y=cummax(y)
y[!is.na(x)]=NA

y
[1] NA NA  2 NA NA  5  5 NA

从Henrik挤到one_line

replace(cummax(seq_along(x) * !is.na(x)), !is.na(x), NA)

Answer 2

这是一种通过tidyr函数fill来获取值的方法

library(tidyr)
fill(as.data.frame(x),x)[is.na(x),]
#[1] 5 4 4