python - 使用 statsmodel 在 Python 中从 GLM 中提取系数

Question

我有一个定义如下的模型:

import statsmodels.formula.api as smf
model = smf.glm(formula="A ~ B + C + D", data=data, family=sm.families.Poisson()).fit()

模型的系数如下所示:

Intercept   0.319813
C[T.foo]   -1.058058
C[T.bar]   -0.749859
D[T.foo]    0.217136
D[T.bar]    0.404791
B           0.262614

我可以通过 model.params.Intercept 和 model.params.B 获取 Intercept 和 B 的值 但我无法获取每个 C 和 D 的值。

例如，我尝试了 model.params.C[T.foo]，但我得到了错误。

我如何从模型中获取特定值？

Answer 1

model.params 是一个 pandas.Series。仅当条目的名称是有效的 Python 名称时，才能作为属性访问。

在这种情况下，您需要使用引号中的名称进行索引，即 model.params["C[T.foo]"]

见 http://pandas.pydata.org/pandas-docs/dev/indexing.html