r - 滞后 a data.frame 的 NA 数量

Question

换句话说，我正在尝试滞后 data.frame，如下所示：

V1 V2 V3 V4 V5 V6 
1  1  1  1  1  1
2  2  2  2  2  NA
3  3  3  3  NA NA
4  4  4  NA NA NA
5  5  NA NA NA NA
6  NA NA NA NA NA

看起来像这样的东西：

V1 V2 V3 V4 V5 V6 
1  NA NA NA NA NA
2  1  NA NA NA NA
3  2  1  NA NA NA
4  3  2  1  NA NA
5  4  3  2  1  NA
6  5  4  3  2  1

到目前为止，我已经使用了一个计算 NA 数量的函数，并试图将 data.frame 中的每一列滞后于该列中相应的 NA 数量。

V1 <- c(1,2,3,4,5,6)
V2 <- c(1,2,3,4,5,NA)
V3 <- c(1,2,3,4,NA,NA)
V4 <- c(1,2,3,NA,NA,NA)
V5 <- c(1,2,NA,NA,NA,NA)
V6 <- c(1,NA,NA,NA,NA,NA)
mydata <- cbind(V1,V2,V3,V4,V5,V6)
na.count <- colSums(is.na(mydata))
lag.by <- function(mydata, na.count){lag(mydata, k = na.count)}
lagged.df <- apply(mydata, 2, lag.by)

但是这段代码只比整个 data.frame 滞后一个...

Answer 1

一个选项是使用 apply 遍历列并首先通过使用 is.na 对 NA 元素进行子集化来附加 NA 元素，然后通过取反逻辑向量 ( is.na)

apply(mydata, 2, function(x) c(x[is.na(x)], x[!is.na(x)]))
#     V1 V2 V3 V4 V5 V6
#[1,]  1 NA NA NA NA NA
#[2,]  2  1 NA NA NA NA
#[3,]  3  2  1 NA NA NA
#[4,]  4  3  2  1 NA NA
#[5,]  5  4  3  2  1 NA
#[6,]  6  5  4  3  2  1

Answer 2

您可以像这样使用带有选项 na.last = FALSE 的 sort 函数：

编辑：

Akrun 的评论是有效的。如果值需要保持在 data.frame 中的顺序，那么 Akrun 的答案是最好的。 Sort 将按照从低到高的顺序排列所有内容，NA 在前。

library(purrr)
map_df(mydata, sort, na.last = FALSE)
# A tibble: 6 x 6
     V1    V2    V3    V4    V5    V6
  <int> <int> <int> <int> <int> <int>
1     1    NA    NA    NA    NA    NA
2     2     1    NA    NA    NA    NA
3     3     2     1    NA    NA    NA
4     4     3     2     1    NA    NA
5     5     4     3     2     1    NA
6     6     5     4     3     2     1

或申请：

apply(mydata, 2, sort , na.last = FALSE)
     V1 V2 V3 V4 V5 V6
[1,]  1 NA NA NA NA NA
[2,]  2  1 NA NA NA NA
[3,]  3  2  1 NA NA NA
[4,]  4  3  2  1 NA NA
[5,]  5  4  3  2  1 NA
[6,]  6  5  4  3  2  1

edit2:

正如尼科洛评论的那样。 order 可以保留变量的顺序。

mydata[,3] <- c(4, 3, 1, 2, NA, NA)
map_df(mydata, function(x) x[order(!is.na(x))])
# A tibble: 6 x 6
     V1    V2    V3    V4    V5    V6
  <int> <int> <dbl> <int> <int> <int>
1     1    NA    NA    NA    NA    NA
2     2     1    NA    NA    NA    NA
3     3     2     4    NA    NA    NA
4     4     3     3     1    NA    NA
5     5     4     1     2     1    NA
6     6     5     2     3     2     1

r - 滞后 a data.frame 的 NA 数量

r - Lag a data.frame by the number of NAs

r

function

lag

apply