根据R中性别的分布类型替换NA

时间:2018-07-13 13:47:38

标签: r dplyr plyr na

当我在此处选择NA值

data[data=="na"] <- NA
data[!complete.cases(data),]

我必须替换它,但要取决于发行类型。 如果使用Shapiro.test通过不正常的变量进行分布, 然后必须用中位数代替缺失值, 如果正常,则用平均值代替。 但是每个性别的分布情况(1个女孩,2个男人)

data=structure(list(sex = c(1L, 1L, 1L, 1L, 2L, 2L, 2L, 2L), emotion = c(20L, 
15L, 49L, NA, 34L, 35L, 54L, 45L), IQ = c(101L, 98L, 105L, NA, 
123L, 120L, 115L, NA)), .Names = c("sex", "emotion", "IQ"), class = "data.frame", row.names = c(NA, 
-8L))

所需的输出

sex emotion IQ
1   20  101
1   15  98
1   49  105
1   28  101
2   34  123
2   35  120
2   54  115
2   45  119

1 个答案:

答案 0 :(得分:2)

以下代码将根据Shapiro测试替换NA值:

library(dplyr)

data %>% 
 group_by(sex) %>%
 mutate(
  emotion = ifelse(!is.na(emotion), emotion,
   ifelse(shapiro.test(emotion)$p.value > 0.05,
    mean(emotion, na.rm=TRUE), quantile(emotion, na.rm=TRUE, probs=0.5) ) ),
  IQ = ifelse(!is.na(IQ), IQ,
   ifelse(shapiro.test(IQ)$p.value > 0.05,
    mean(IQ, na.rm=TRUE), quantile(IQ, na.rm=TRUE, probs=0.5) )
  )
 )