多项式回归python代码
时间: 2023-07-03 21:04:25 浏览: 95
以下是使用Python进行多项式回归的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成样本数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.3, 4.5, 6.7, 8.9, 11.1])
# 多项式回归
z = np.polyfit(x, y, 2)
p = np.poly1d(z)
# 绘制图像
xp = np.linspace(1, 5, 100)
plt.plot(x, y, '.', xp, p(xp), '-')
plt.show()
```
在上述代码中,首先生成了5个样本数据,然后使用`np.polyfit()`函数进行多项式回归,其中第三个参数为多项式的阶数。最后使用`np.poly1d()`函数将回归结果转化为一个多项式对象,方便后续使用。最后使用`plt.plot()`函数将原始数据点和回归曲线绘制在图像上。
阅读全文