twopm包margin/AME函数报错：数据框列未定义或变量缺失

问题描述 #

用R语言twopm包构建两部分模型tpmodel1后，调用margin()或AME()函数计算边际效应时出现异常：

• 直接运行margin(tpmodel1)或指定newdata=NULL时，报错：Error in [.data.frame(newdata, , term) : undefined columns selected
• 指定newdata为建模原数据集dfasthma时，报错：some of independent variables of the two-part model are not available in the newdata set，但dfasthma确实是建模所用数据集。

报错代码示例 #

> margin(tpmodel1)
Error in `[.data.frame`(newdata, , term) : undefined columns selected

> margin(tpmodel1, newdata = NULL, term = NULL, se=TRUE)
Error in `[.data.frame`(newdata, , term) : undefined columns selected

> margin(tpmodel1, newdata = dfasthma, term = NULL, se=TRUE)
Error in margin(tpmodel1, newdata = dfasthma, term = NULL, se = TRUE) : 
some of independent variables of the two-part model are not available in the newdata set

margin函数语法 #

margin(object, newdata = NULL, term = NULL, value = NULL,
se = TRUE, se.method = c("delta","bootstrap"), CI = TRUE, CI.boots = FALSE,
level = 0.95,eps = 1e-7,na.action = na.pass, iter = 50)

排查与解决步骤 #

• 核对模型变量与数据集一致性：用names(tpmodel1$first$model)和names(tpmodel1$second$model)分别查看两部分模型的变量，再和names(dfasthma)对比，确认是否存在变量名不一致（如大小写、空格）或建模时临时衍生的变量（如scale(age)、factor(gender)）未保留在原数据集的情况。若有衍生变量，需提前将其存入dfasthma。
• 显式指定term参数：不要用term=NULL，手动指定模型中的变量名，例如margin(tpmodel1, newdata=dfasthma, term=c("age", "gender"))，避免函数自动识别变量出错。
• 检查数据集完整性：用complete.cases(dfasthma)排查缺失值，或用na.omit(dfasthma)处理后传入函数，同时调整na.action参数（如na.action=na.omit）适配数据情况。
• 规范建模流程：建模时避免在公式中临时衍生变量，提前在数据集内创建所需变量（如先执行dfasthma$scaled_age <- scale(dfasthma$age)，再用twopm(y ~ scaled_age, data=dfasthma)建模），确保newdata包含模型所需全部变量。
• 尝试AME函数替代调用：若margin仍有问题，试调用AME(tpmodel1, data=dfasthma)，显式传入数据集并指定变量。

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