从上一个非 NA 值获取索引

Get index from last previous non-NA value

我有这个向量

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

基础 Rtidyverse 都可以。我尝试了 lagleadrleglcoalesce 的组合,但都没有成功。例如这很接近,但仍然是错误的。

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

这是使用 tidyr 函数 fill

获取值的方法
library(tidyr)
fill(as.data.frame(x),x)[is.na(x),]
#[1] 5 4 4

使用基数 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

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