python拟合曲线函数代码
时间: 2023-11-12 07:09:22 浏览: 93
拟合曲线代码
好的,以下是一个使用Python进行曲线拟合的示例代码:
```python
import numpy as np
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt
# 定义拟合函数
def func(x, a, b, c):
return a * np.sin(b * x) + c
# 生成数据
x = np.linspace(0, 10, 100)
y = func(x, 1, 1.5, 0.5) + np.random.randn(100)
# 执行曲线拟合
popt, pcov = curve_fit(func, x, y)
# 打印拟合函数的参数
print(popt)
# 画出原始数据和拟合结果
plt.plot(x, y, 'o', label='data')
plt.plot(x, func(x, *popt), label='fit')
plt.legend()
plt.show()
```
以上代码中,我们使用 `curve_fit` 函数进行曲线拟合,需要定义拟合函数并传入要拟合的数据。在示例代码中,我们生成了一个带有误差的样本,并使用 `func` 函数进行正弦曲线的拟合。最终,我们将拟合后的曲线和原始数据一起画在图中以进行比较。
阅读全文