Замена NA в зависимости от типа распределения пола в R

Когда я выбрал значение 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

r dplyr na plyr
person varimax    schedule 13.07.2018    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Следующий код заменит значения 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) )
  )
 )
person Lstat    schedule 13.07.2018