python 多项式求系数_python – sklearn：如何获得多项式特征的系数

顺便说一句,现在有更合适的功能：

PolynomialFeatures.get_feature_names.

from sklearn.preprocessing import PolynomialFeatures

import pandas as pd

import numpy as np

data = pd.DataFrame.from_dict({

'x': np.random.randint(low=1, high=10, size=5),

'y': np.random.randint(low=-1, high=1, size=5),

})

p = PolynomialFeatures(degree=2).fit(data)

print p.get_feature_names(data.columns)

这将输出如下：

['1', 'x', 'y', 'x^2', 'x y', 'y^2']

注：出于某种原因,您必须先使用PolynomialFeatures对象,然后才能使用get_feature_names().

如果您是Pandas-lover(就像我一样),您可以轻松地使用以下所有新功能形成DataFrame：

features = DataFrame(p.transform(data), columns=p.get_feature_names(data.columns))

print features

结果将如下所示：

1 x y x^2 x y y^2

0 1.0 8.0 -1.0 64.0 -8.0 1.0

1 1.0 9.0 -1.0 81.0 -9.0 1.0

2 1.0 1.0 0.0 1.0 0.0 0.0

3 1.0 6.0 0.0 36.0 0.0 0.0

4 1.0 5.0 -1.0 25.0 -5.0 1.0