pycharm拟合曲线
时间: 2023-11-16 14:02:17 浏览: 126
机器学习实验基于Python实现多项式拟合正弦曲线源代码+实验报告,代码含详细解释
PyCharm是一款常用的Python集成开发环境(IDE),可以用于编写、调试和运行Python程序。在PyCharm中,可以使用Matplotlib库绘制曲线,并使用Scipy库中的最小二乘法进行曲线拟合。下面是一个简单的曲线拟合的例子:
```python
import numpy as np
from scipy.optimize import leastsq
from matplotlib import pyplot as plt
# 生成数据
x_data = np.linspace(0, 10, 100)
y_data = 2 * x_data + 1 + np.random.randn(100)
# 绘制原始数据
plt.plot(x_data, y_data, 'bo', label='data')
# 定义拟合函数形式
def Fun(p, x):
a1, a2 = p
return a1 * x + a2
# 定义拟合残差
def error(p, x, y):
return Fun(p, x) - y
# 使用最小二乘法拟合曲线
p0 = np.random.randn(2) # 拟合的初始参数设置
para = leastsq(error, p0, args=(x_data, y_data)) # 进行拟合
a1, a2 = para[0]
# 生成拟合曲线数据
x_fit = np.linspace(0, 10, 100)
y_fit = a1 * x_fit + a2
# 绘制拟合曲线
plt.plot(x_fit, y_fit, 'r-', label='fit')
# 添加图例
plt.legend()
# 显示图像
plt.show()
```
在这个例子中,我们首先生成了一组随机数据,然后使用最小二乘法拟合了一条直线,并将原始数据和拟合曲线绘制在同一张图上。
阅读全文