python - 使用 pandas 和 statsmodels 时提取变量名

Question

我正在尝试从 statsmodels 生成的结果中访问变量的名称。我将在示例代码之后详细说明。

import scikits.statsmodels.api as sm
import pandas as pd
data = sm.datasets.longley.load()
df = pd.DataFrame(data.exog, columns=data.exog_name)
y = data.endog
df['intercept'] = 1.
olsresult = sm.OLS(y, df).fit()

此摘要输出包括变量名称。当您调用类似 olsresult.params 的内容时，它会返回以下内容:

In [21]: olsresult.params
Out[21]: 
GNPDEFL           15.061872
GNP               -0.035819
UNEMP             -2.020230
ARMED             -1.033227
POP               -0.051104
YEAR            1829.151465
intercept   -3482258.634596

现在我很好奇的是创建一个类似于字典的东西，其中变量名作为键，参数值作为值。所以，像 {'GNPDELF':15.0618, 'GNP':-0.035819} 等等。如果无法做到这一点，是否有任何其他方法可以单独访问变量名称和值？

Answer 1

尝试显而易见的事情总是值得的。:^)

In [14]: olsresult.params
Out[14]: 
GNPDEFL           15.061872
GNP               -0.035819
UNEMP             -2.020230
ARMED             -1.033227
POP               -0.051104
YEAR            1829.151465
intercept   -3482258.634597

In [15]: dict(olsresult.params)
Out[15]: 
{'ARMED': -1.0332268671737328,
 'GNP': -0.035819179292614578,
 'GNPDEFL': 15.061872271452557,
 'POP': -0.051104105653539733,
 'UNEMP': -2.0202298038172479,
 'YEAR': 1829.151464613984,
 'intercept': -3482258.6345966831}

另请参阅 Series 对象的 .to_dict() 方法。