将NA重新插入向量中

Question

我有一个包含NA的值向量.这些值需要由无法处理NA的外部程序处理,因此它们被剥离,写入文件,处理,然后读回,产生非NA数量长度的向量.例如,假设输入为7 3 4 NA 5 4 6 NA 1 NA,那么输出将只是7个值.我需要做的是将NA重新插入到位.

因此,给定两个向量X和Y:

 > X
 [1]  64   1   9 100  16  NA  25  NA   4  49  36  NA  81
 > Y
 [1]  8  1  3 10  4  5  2  7  6  9

生产:

8 1 3 10 4 NA 5 NA 2 7 6 NA 9

(你可能会注意到X是Y ^ 2,这只是一个例子).

我可以敲出一个函数来做这个,但我想知道是否有任何好的狡猾的方法做到这一点...分裂,列表,长度...嗯......

Answer 1

na.omit保留NA原始系列中位置的属性,因此您可以使用它来知道缺少值的位置:

Y <- sqrt(na.omit(X))
Z <- rep(NA,length(Y)+length(attr(Y,"na.action")))
Z[-attr(Y,"na.action")] <- Y
#> Z
# [1]  8  1  3 10  4 NA  5 NA  2  7  6 NA  9

Answer 2

同一主题的另一个变体

rena <- function(X,Z){
    X[which(!is.na(X))]=Z    
    X
}

R 自动用 NA 填充其余部分。

编辑：由马雷克纠正。