基于因子水平的估算

时间:2018-02-14 03:19:05

标签: r r-mice

我正在尝试使用MICE对以下数据框进行估算。

marketValue <- c(NA, 234234, NA, 243243, NA, NA, 234523, NA, 232427, 112214)
bathrooms <- c(3,3,2,3,5,4,1,5,6,3)
garageSqFt <- c(400, 385, 454, 534, 210, NA, 342, 423, 535, NA)
totalSqFT <- c(NA, NA, 1231, 2232, 4564, 2122, 4324, 4342, 1299, 4355)
units <- c(1, 1, 1, 1, 1, 1, 1.5, NA, 2, 5)
subDivId <- c("112", "111", "111", "111", "112", "111", "112", "112", "111", 
"112")
data <- data.frame(marketValue, bathrooms, garageSqFt, totalSqFT, units, 
subDivId)

在实际数据框中,大约有1300个subDivId因子级别,我想创建新的数据帧(每个数据帧都有相同的subDivId行),然后在每个数据帧中输入。我的尝试

splitSubDiv <- split(data, data$subDivId) 
for (neighborhood in splitSubDiv){
    testData <- mice(neighborhood, m=5, maxit = 5) 
    impData <- complete(testData, 5) 
    str(impData) }

这似乎不起作用。没有任何估算,事情仍然是NA。我做错了什么?

1 个答案:

答案 0 :(得分:0)

在您的尝试中,您没有将估算的数据框分配给对象,因此不会保留结果。

您可以使用lapply而不是for循环来尝试此方法。您最终会得到一个插补数据框列表。

splitSubDivImp <- lapply(splitSubDiv, function(neighborhood) {
  testData <- mice(neighborhood, m = 5, maxit = 5) 
  impData <- complete(testData, 5)
})

splitSubDivImp