添加statsmodels'预测'结果为Pandas数据帧

时间:2014-03-22 16:43:26

标签: pandas dataframe linear-regression prediction statsmodels

通常希望将预测结果附加到用于进行预测的数据集中,但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返回一个保留其参数索引的数据帧?

1 个答案:

答案 0 :(得分:2)

预测不应该丢弃任何行。你能发布一个最小的工作示例吗?保留熊猫指数在我的雷达上,应尽快修好。

https://github.com/statsmodels/statsmodels/issues/1501

修改:没关系。这是一个已知的问题。 https://github.com/statsmodels/statsmodels/issues/1352