python statsmodels - 回归中的二次项

Question

我有以下线性回归:

import statsmodels.formula.api as sm

model = sm.ols(formula = 'a ~ b + c', data = data).fit()

我想在这个模型中为 b 添加一个二次项。

statsmodels.ols 有简单的方法吗？
我应该使用更好的软件包来实现这一目标吗？

Answer 1

最简单的方法是

model = sm.ols(formula = 'a ~ b + c + I(b**2)', data = data).fit()

I(...) 基本上是在说“patsy，请不要在这里自作多情，让 Python 处理 kthx 中的所有内容”。 ( More detailed explanation )