Могу ли я извлечь или построить в виде Pandas таблицы данных со значениями коэффициентов et c. предоставляется методом summary () в statsmodels? - PullRequest
Новая
       87

Могу ли я извлечь или построить в виде Pandas таблицы данных со значениями коэффициентов et c. предоставляется методом summary () в statsmodels?

0 голосов
/ 12 января 2020

Я запустил модель OLS в statsmodels и хотел бы, чтобы таблица в сводке представляла собой Pandas фрейм данных.

Это то, что я имею в виду:

enter image description here

Я бы хотел, чтобы таблица в красной рамке была построена / извлечена и стала Pandas DataFrame.

Мой код до этого момента был простым: 

from statsmodels.regression.linear_model import OLS

mod = OLS(endog = coded_design_poly_select.response.values, exog = coded_design_poly_select.iloc[:, :-1].values)

fitted_model = mod.fit()

fitted_model.summary()

Что бы вы предложили?

1 Ответ

1 голос
/ 14 января 2020

fitted_model на самом деле является RegressionResults объектом, который хранит все результаты регрессии, и вы можете получить к ним доступ через соответствующие методы / атрибуты .

Для того, что вы просили Я полагаю, что следующий код будет работать 

data = {'coef': fitted_model.params,
        'std err': fitted_model.bse,
        't': fitted_model.tvalues,
        'P>|t|': fitted_model.pvalues,
        '[0.025': fitted_model.conf_int()[0],
        '0.975]': fitted_model.conf_int()[1]}

pd.DataFrame(data).round(3)
...