python画图拟合曲线
时间: 2023-10-30 20:07:53 浏览: 107
python-曲线拟合-原理-代码.doc
要在Python中画出拟合曲线,可以使用matplotlib和numpy库。首先,导入这两个库。然后,设置x和y的值。接下来,使用polyfit函数通过拟合多项式来计算拟合曲线的参数。最后,使用plot函数绘制原始数据点和拟合曲线,并使用show函数显示图形。
下面是一个示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 1, 25)
y = np.cos(x) + 0.3 * np.random.rand(25)
= np.poly1d(np.polyfit(x, y, 4))
t = np.linspace(0, 1, 250)
plt.plot(x, y, 'o', t, p(t), '-')
plt.show()
```
这段代码首先使用linspace函数生成一个包含25个均匀分布的点的数组x。然后,使用cos函数生成与x对应的y值,并添加一些随机噪声。接下来,使用polyfit函数拟合出一个4次多项式,并将其赋给p。最后,使用plot函数绘制出原始数据点和拟合曲线。
另外,你还可以使用curve_fit函数进行曲线拟合。这个函数使用的是非线性最小二乘法来拟合给定的函数模型。下面是一个示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
x = np.arange(0, 20)
y = 2 * x ** 2 + np.random.randint(0, 100, 20)
def func(x, a, b, c):
return a * x**2 + b * x + c
popt, pcov = curve_fit(func, x, y)
y2 = func(x, popt
阅读全文