R中使用mice多重插补得到mids类数据后如何提取指定列？

推荐方案：直接通过mice内置方法批量建模（无需手动拆分数据集） #

mice包原生支持对mids类插补结果批量调用建模函数，不需要手动拆分5个数据集再逐一提列，能避免手动处理的疏漏，后续也方便直接合并插补建模结果。
示例代码如下，替换成你自己的变量名和rrr参数即可：

# 加载依赖包
library(mice)
library(rrr) # 替换成你实际使用的提供rrr函数的包

# 假设你的mids类多重插补结果存储在对象imp中
mi_rrr_result <- with(
  data = imp,
  expr = rrr(
    # 直接写需要作为响应变量的列名，支持多响应列用cbind拼接
    y = cbind(你的响应变量1, 你的响应变量2, 你的响应变量3),
    # 直接写需要作为预测变量的列名，多预测变量同样用cbind拼接
    x = cbind(你的预测变量1, 你的预测变量2, 你的预测变量3, 你的预测变量4),
    # 补充你需要设置的其他rrr模型参数，比如秩约束、惩罚项等
    rank = 2
  )
)

with方法会自动遍历5个插补完成的数据集，自动提取你指定的列传入rrr函数，返回包含5个插补集模型结果的对象，后续可按多重插补的分析规范合并结果。

手动提取列方案（适合需要自定义预处理的场景） #

如果你确实需要单独拆分出每个插补集的响应变量、预测变量做自定义处理，可以用complete()函数先把mids对象转成完整数据集列表，再按列名提取目标列：

# 导出全部5个插补完成的数据集，存储为列表，每个元素对应1个插补集
imp_dataset_list <- complete(imp, action = "all")

# 定义你需要提取的列名，替换成你数据集里的实际列名
response_col_names <- c("响应变量1", "响应变量2", "响应变量3")
predictor_col_names <- c("预测变量1", "预测变量2", "预测变量3", "预测变量4")

# 批量提取每个插补集的响应变量、预测变量，分别存为列表
response_list <- lapply(imp_dataset_list, function(dt) dt[, response_col_names])
predictor_list <- lapply(imp_dataset_list, function(dt) dt[, predictor_col_names])

# 逐一对每个插补集的提取结果拟合rrr模型
rrr_fit_list <- mapply(
  function(y_data, x_data) {
    rrr(y = as.matrix(y_data), x = as.matrix(x_data), rank = 2)
  },
  y_data = response_list,
  x_data = predictor_list,
  SIMPLIFY = FALSE
)

注意事项

• 提取列优先使用列名索引，不要用数字位置索引，避免数据集列顺序变动导致提取错误
• 若你使用的rrr函数要求输入为矩阵格式，需要参照示例用as.matrix()把提取的data.frame列转换格式
• 不要仅使用单个插补集的结果做统计推断，必须按规则合并5个插补集的模型结果，否则会低估参数标准误

内容的提问来源于stack exchange，提问作者zean song