从上一个非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

基地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)