python多项式线性回归算法代码
时间: 2023-12-06 21:38:40 浏览: 100
线性回归算法代码.zip
5星 · 资源好评率100%
以下是Python中实现多项式回归的代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.array([0.0, 1.0, 2.0, 3.0, 4.0, 5.0])
y = np.array([0.1, 0.9, 2.2,2.8, 3.6, 5.0])
# 多项式回归
z = np.polyfit(x, y, 3) # 3表示多项式的次数
p = np.poly1d(z)
# 绘制图像
xp = np.linspace(-1, 6, 100)
plt.plot(x, y, '.', xp, p(xp), '-')
plt.show()
```
这段代码首先生成了一些数据,然后使用`np.polyfit()`函数进行多项式回归,其中第一个参数是自变量,第二个参数是因变量,第三个参数是多项式的次数。然后使用`np.poly1d()`函数将回归结果转换为一个多项式对象,最后使用`plt.plot()`函数绘制图像。
阅读全文