patsy公式-增加因子的幂

Question

我用patsy来建立设计矩阵。我需要包括原始因素的力量。例如，使用回归

，我想能写

patsy.dmatrix('y~x1 + x1**2 + x2 + x2**2 + x2**3', 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