通常希望将预测结果附加到用于进行预测的数据集中,但statsmodels predict
函数返回(非索引)结果的长度可能与预测的数据集不同是基于。
例如,如果测试数据集test
包含任何空条目,则
mod_fit = sm.Logit.from_formula('Y ~ A B C', train).fit()
press = mod_fit.predict(test)
将生成一个比test
长度短的数组,并且无法有效地附加
test['preds'] = preds
由于predict
的结果未编入索引,因此无法恢复应附加结果的行。
将predict
结果与生成它们的行相关联的惯用法是什么?是否有一种方法可以让predict
返回一个保留其参数索引的数据帧?
答案 0 :(得分:2)
预测不应该丢弃任何行。你能发布一个最小的工作示例吗?保留熊猫指数在我的雷达上,应尽快修好。
https://github.com/statsmodels/statsmodels/issues/1501
修改:没关系。这是一个已知的问题。 https://github.com/statsmodels/statsmodels/issues/1352