python画拟合曲线
时间: 2023-11-12 21:59:57 浏览: 92
基于Python和numpy计算出拟合数据并画出拟合曲线(源码).rar
5星 · 资源好评率100%
好的,关于Python画拟合曲线,可以使用matplotlib库中的plot函数来实现。具体步骤如下:
1. 导入需要的库
```python
import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
```
2. 定义拟合函数
```python
def func(x, a, b, c):
return a * np.exp(-b * x) + c
```
这里以指数函数为例,可以根据实际情况定义不同的函数。
3. 生成数据
```python
xdata = np.linspace(0, 4, 50)
y = func(xdata, 2.5, 1.3, 0.5)
ydata = y + 0.2 * np.random.normal(size=len(xdata))
```
这里生成了50个数据点,其中ydata加入了一些噪声。
4. 拟合数据
```python
popt, pcov = curve_fit(func, xdata, ydata)
```
这里使用curve_fit函数进行拟合,popt为拟合参数,pcov为协方差矩阵。
5. 绘制拟合曲线
```python
plt.plot(xdata, ydata, 'b-', label='data')
plt.plot(xdata, func(xdata, *popt), 'r-', label='fit')
plt.legend()
plt.show()
```
这里使用plot函数绘制原始数据和拟合曲线,并使用legend函数添加图例。
阅读全文