R - получение вмененных пропущенных значений обратно в фрейм данных

Я использую aregImpute для вменения отсутствующих значений в кадр данных R (bn_df).

Код такой:

library(Hmisc)
impute_arg <- aregImpute(~ TI_Perc + AS_Perc + 
                         CD_Perc + CA_Perc + FP_Perc, 
                         data = bn_df, n.impute = 5)

Это работает нормально.

Проблема после. При возвращении значений в исходный фрейм данных.

Я могу это сделать, только не очень элегантно. В основном мне нужно скопировать/вставить следующую строку для всех столбцов:

bn_df$CD_Perc[impute_arg$na$CD_Perc] <- impute_arg$imputed$CD_Perc[,1]
bn_df$FP_Perc[impute_arg$na$FP_Perc] <- impute_arg$imputed$FP_Perc[,1]
...

Это работает. Но должен быть более эффективный способ сделать это без копирования/вставки для всех столбцов.

Любые идеи?


r imputation hmisc
person BrunoPT    schedule 02.02.2017    source источник

Ответы (1)


arrow_upward
7
arrow_downward

Вы можете использовать функцию impute.transcan. Поскольку вы не предоставили данные, я скопировал пример из документации aregImpute.

# The data
x1 <- factor(sample(c('a','b','c'),1000,TRUE))
x2 <- (x1=='b') + 3*(x1=='c') + rnorm(1000,0,2)
x3 <- rnorm(1000)
y  <- x2 + 1*(x1=='c') + .2*x3 + rnorm(1000,0,2)
orig.x1 <- x1[1:250]
orig.x2 <- x2[251:350]

# Insert NAs
x1[1:250] <- NA
x2[251:350] <- NA

# Create a data frame 
d <- data.frame(x1,x2,x3,y)
# Find value of nk that yields best validating imputation models
# tlinear=FALSE means to not force the target variable to be linear

# Use imputation 
f <- aregImpute(~y + x1 + x2 + x3, nk=c(0,3:5), tlinear=FALSE,
                data=d, B=10) # normally B=75

# Get the imputed values
imputed <-impute.transcan(f, data=d, imputation=1, list.out=TRUE, pr=FALSE, check=FALSE)

# convert the list to the database
imputed.data <- as.data.frame(do.call(cbind,imputed))

# arrange the columns accordingly
imputed.data <- imputed.data[, colnames(d), drop = FALSE]
person discipulus    schedule 03.02.2017
comment
Благодарю вас! Это сработало отлично. Действительно, я, вероятно, пропустил эту функцию. - person BrunoPT; 03.02.2017
comment
Можем ли мы каким-либо образом заменить значение средним значением нескольких вмен, а не каким-либо конкретным числом (вменение = 1)? - person Sourabh; 13.04.2017