将单元格值分配给后续 NA 单元格的函数（同一列）

Question

感谢您抽空查看我的问题！我是论坛的新手，也是 R 的新手，但我会尽力把问题表述清楚。

我有一大组树样本数据，每个个体的行数不规则。在 "class" 变量列（此处为第 2 列）中，每个个体的第一行都有一个值（1、2、3 或 4），随后的值为 NA。我想将每个人的第一个值分配给相应的后续 NA 单元格（属于同一个人）。

可重现的示例数据框（已编辑）：

test <- cbind(c(1, 2, 3, NA, 4, NA, NA, NA, 5, NA, 6), c(3, 4, 3, NA, 1, NA, NA, NA, 2, NA, 1))
colnames(test) <- c("ID", "class")

        ID  class
 [1,]    1    3
 [2,]    2    4
 [3,]    3    3
 [4,]   NA   NA
 [5,]    4    1
 [6,]   NA   NA
 [7,]   NA   NA
 [8,]   NA   NA
 [9,]    5    2
[10,]   NA   NA
[11,]    6    1

我要找的结果是这样的：

      ID class
 [1,]  1     3
 [2,]  2     4
 [3,]  3     3
 [4,] NA     3
 [5,]  4     1
 [6,] NA     1
 [7,] NA     1
 [8,] NA     1
 [9,]  5     2
[10,] NA     2
[11,]  6     1

我复制了这个主题的最后一个解决方案How to substitute several NA with values within the DF using if-else in R? 并尝试像这样适应我的需要

    test2 <- as.data.frame(t(apply(test["class"], 1, function(x)
    if(is.na(x[1]) == FALSE && all(is.na(head(x[1], -1)[-1])))
    replace(x, is.na(x), x[1]) else x)))

但它给了我错误 "dim(x) must have positive length"。我尝试了许多其他版本，它给了我各种各样的错误，我什至不知道从哪里开始。我该如何改进它？

Answer 1

如果您不想加载另一个包，这里有一个小的单行函数可以使用：

rollForward <- function(x) {
    c(NA, x[!is.na(x)])[cumsum(!is.na(x)) + 1]
}

test[,"class"] <- rollForward(test[,"class"])
test
#       ID class
#  [1,]  1     3
#  [2,]  2     4
#  [3,]  3     3
#  [4,] NA     3
#  [5,]  4     1
#  [6,] NA     1
#  [7,] NA     1
#  [8,] NA     1
#  [9,]  5     2
# [10,] NA     2
# [11,]  6     1

将单元格值分配给后续 NA 单元格的函数（同一列）

Function to assign cell value to subsequent NA-cells (same column)

if-statement

r

na