patsy公式 - 增加一个因子的权力

Question

我用patsy构建设计矩阵。我需要包括原始因素的权力。例如，使用回归，我希望能够编写

patsy.dmatrix('y~x1 + x1**2 + x2 + x2**2 + x2**3', data)

其中data是包含y，x1，x2列的数据帧。但它似乎根本不起作用。任何解决方案？

Answer 1

Patsy对**有一个特殊的解释，它继承自R.我曾考虑让它在应用于数字因子时自动做正确的事情，但实际上没有实现它......时间，有一种通用方法告诉patsy切换到使用Python解释操作符，而不是Patsy解释：你将表达式包装在I(...)中。所以：

patsy.dmatrix('y~x1 + I(x1**2) + x2 + I(x2**2) + I(x2**3)', data)

（More detailed explanation here）

Answer 2

Patsy似乎没有管理权力代表（但是？）。您可以在此处找到一种出行方式：python stats models - quadratic term in regression