python拟合多项式回归函数代码
时间: 2023-08-29 19:07:03 浏览: 153
多项式拟合 示例代码 python
当然可以帮您解决这个问题,以下是Python拟合多项式回归函数代码的示例:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成随机数据
x = np.linspace(-5, 5, 50)
y = x**3 - 5*x**2 + 4*x + 2 + np.random.randn(len(x)) * 10
# 多项式回归
z = np.polyfit(x, y, 3) # 拟合3次多项式回归
p = np.poly1d(z) # 构造回归函数
# 可视化
xp = np.linspace(-5, 5, 100)
plt.plot(x, y, '.', xp, p(xp), '-r')
plt.show()
```
以上代码通过`np.polyfit()`函数实现多项式回归,`np.poly1d()`函数构造回归函数,可视化部分使用了`matplotlib`库。
阅读全文