在Python中,你可以使用`sklearn`库中的`PolynomialFeatures`类来生成多项式特征,或者自己编写函数来实现多项式的计算。以下是使用`sklearn`生成多项式特征的示例代码:
import numpy as np
from sklearn.preprocessing import PolynomialFeatures
生成三行二列的数组
x = np.arange(6).reshape(3, 2)
设置多项式的度数为3
reg = PolynomialFeatures(degree=3)
将数据转换为多项式特征
x_poly = reg.fit_transform(x)
print("n=3时,sklearn的输出是:")
print(x_poly)
如果你想要自己实现多项式计算,可以使用以下函数:
def multi_feature(x, n):
c = np.empty((x.shape, 0)) 创建一个空数组,行数为x的行数,列数为0
for i in range(n + 1):
for m in range(i, -1, -1):
h = (x[:, 0] m) * (x[:, 1] (i - m))
c = np.c_[c, h]
return c
print("n=4时,函数的输出是:")
print(multi_feature(x, 4))
以上代码展示了如何使用`sklearn`和自定义函数来生成多项式特征。你可以根据需要调整多项式的度数`degree`。