多项式拟合Python
时间: 2023-08-14 13:11:30 浏览: 111
基于Python实现的多项式拟合曲线实验.zip
5星 · 资源好评率100%
多项式拟合是一种常见的数据拟合方法,可以使用Python的NumPy库来进行实现。下面是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成一些示例数据
x = np.linspace(0, 10, 100)
y = 2 * x**2 - 3 * x + 1 + np.random.randn(100)
# 进行多项式拟合
degree = 2 # 多项式的次数
p = np.polyfit(x, y, degree)
# 根据拟合结果生成拟合曲线
x_fit = np.linspace(0, 10, 100)
y_fit = np.polyval(p, x_fit)
# 绘制原始数据和拟合曲线
plt.scatter(x, y, label='Original Data')
plt.plot(x_fit, y_fit, color='red', label='Polyfit Curve')
plt.legend()
plt.show()
```
在这个示例中,我们首先生成了一些示例数据 `x` 和 `y`。然后使用 `np.polyfit` 函数进行多项式拟合,其中 `degree` 参数指定了多项式的次数。拟合结果保存在变量 `p` 中。接着,我们使用 `np.polyval` 函数根据拟合结果生成拟合曲线的 `x_fit` 和 `y_fit` 值。最后,使用 `matplotlib.pyplot` 库绘制原始数据和拟合曲线。
你可以根据需要调整多项式的次数和数据集,来进行多项式拟合的实验。
阅读全文