如何在python中找到基本展开矩阵

Question

我正在进行非参数回归，需要一个函数将设计矩阵X扩展为基本矩阵。是否有可以执行此操作的软件包？

例如，如果X为200*10（200 obs和10个要素），则使用具有5个碱基的B样条基展开将产生200*50基矩阵。

我尝试过scipy.interpolate.BSpline，但似乎它不返回基矩阵。

Answer 1

python中的粘贴库很有用。

from pasty import dmatrix

transformed_x = dmatrix(
                    "bs(x, df=df, degree=degree, include_intercept=False)",
                    {"train": x}, return_type='matrix')

这将返回向量x的基本展开。如果您有数据矩阵，请对每列执行此操作。