python - statsmodels 示例似乎不起作用

Question

import statsmodels.formula.api as sm
import numpy as np
import pandas

url = "http://vincentarelbundock.github.com/Rdatasets/csv/HistData/Guerry.csv"
df = pandas.read_csv(url)
df = df[['Lottery', 'Literacy', 'Wealth', 'Region']].dropna()
print df.head()
mod = sm.ols(formula='Lottery ~ Literacy + Wealth + Region', data=df)
res = mod.fit()
print res.summary()

打印表格后返回此错误。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-4-f69caff21ed0> in <module>()
6 df = df[['Lottery', 'Literacy', 'Wealth', 'Region']].dropna()
7 print df.head()
----> 8 mod = sm.ols(formula='Lottery ~ Literacy + Wealth + Region', data=df)
9 res = mod.fit()
10 print res.summary()

TypeError: from_formula() takes at least 3 arguments (2 given)

这似乎不是可接受的行为。我做错了什么？

Answer 1

(我评论中的猜测是错误的)

您的 statsmodels 版本太旧。对于 statsmodels 0.5 的发布版本，文档和示例是正确的。

自 0.5.0.dev-1bbd4ca 起，data 关键字已从 df 重命名。

因此，要么升级(我强烈推荐)，要么使用旧的关键字名称

mod = sm.ols(formula='Lottery ~ Literacy + Wealth + Region', df=df)

应该适用于您拥有的版本。