使用R中的二项式分布估算缺失值

时间:2018-07-23 11:37:40

标签: r dplyr spss imputation

我有一列缺少值(q1 = 9),我想根据q1 = 1(=是)和q1 = 2(= no)二项式分布来估算它,如下面的SPSS脚本。我找不到R等效函数

SPSS代码:

SPSS version :IF  q_1 = 9 x=RV.BINOM(1,0.976) .
if q_1 = 9 & x=1 q_1 = 1.
if q_d1 = 9 & x=0 q_1 = 2.

列摘要是这样的

 q_1    n    percent
    1 5868   97.56%
    2  142   2.36%
    9    5   0.08%

2 个答案:

答案 0 :(得分:2)

您可以使用sample生成估算值。

Missing = which(q1 == 9)
q1[Missing] = sample(2, length(Missing), prob=c(0.976, 0.024))

答案 1 :(得分:1)

那呢:

library(tidyverse)

vect1 <- runif(10000, 0, 1)
vect1a <- case_when(
  vect1 < 0.9756 ~ 1,
  vect1 < 0.9756 + 0.0236 ~ 2,
  TRUE ~ 9)
df1 <- tibble(q1 = vect1a)

pct1 <- 0.9756 / (1 - 0.008)
df1a <-  df1 %>% 
  mutate(rand_id = runif(nrow(.), 0, 1),
         q1a = case_when(q1 < 9 ~ q1,
                         rand_id < pct1 ~ 1,
                         TRUE ~ 2))

鼠标包也可以处理归因缺失值,但不能确定它是否使用二项式分布。