r - 使用基于 ID 变量的因子值填充缺失值

Question

我想填写<NA>具有基于 ID 的正确因子值多变的。

以下是变量:

ID <- c(1,1,1,2,2,2,3,3,3)
Gender_NA <- c("m",NA,"m",NA,"f",NA,"m","m",NA)
Gender  <- c("m","m","m","f","f","f","m","m","m")

这是我拥有的数据:

Data_have <- data.frame (ID,Gender_NA)

ID    Gender_NA
 1     m
 1    <NA>
 1     m
 2    <NA>
 2     f
 2    <NA>
 3     m
 3     m
 3    <NA>

这是我想要的数据:

Data_whant <- data.frame (ID,Gender)

ID Gender
1    m
1    m
1    m
2    f
2    f
2    f
3    m
3    m
3    m

我试图在这个论坛上找到解决方案，但我无法让我工作。

非常感谢您的帮助。

Answer 1

na.locf来自 library(zoo) 的函数可用于替换NA元素由相邻的非 NA 前一个元素组成。使用data.table ，我们将'data.frame'转换为'data.table'，按'ID'分组，我们将NA元素替换为前面的非NA，如果第一个元素是NA，则不会被替换，我们可以使用第二个na.locf选项 fromLast=TRUE用后续的非 NA 元素替换剩余的 NA。

library(zoo)
library(data.table)
setDT(Data_have)[, Gender := na.locf(na.locf(Gender_NA, 
            na.rm=FALSE),fromLast=TRUE), by = ID][, Gender_NA := NULL]
Data_have
#    ID Gender
#1:  1      m
#2:  1      m
#3:  1      m
#4:  2      f
#5:  2      f
#6:  2      f
#7:  3      m
#8:  3      m
#9:  3      m

或者按 ID 分组时，我们可以使用 na.omit() 省略所有 NA并选择第一个元素，如下所示:

setDT(Data_have)[, Gender := na.omit(Gender_NA)[1L], by =  ID][, Gender_NA := NULL]

---

或者使用与 dplyr 相同的方法:

library(dplyr)
Data_have %>% 
     group_by(ID) %>%
     transmute(Gender= first(na.omit(Gender_NA)))
#    ID Gender
#   (dbl) (fctr)
#1     1      m
#2     1      m
#3     1      m
#4     2      f
#5     2      f
#6     2      f
#7     3      m
#8     3      m
#9     3      m