python散点图拟合曲线
时间: 2023-07-08 09:47:32 浏览: 222
Python实现numpy计算出拟合数据并画出拟合曲线完整源码+数据.zip
要绘制散点图并拟合曲线,可以使用Python中的Matplotlib和NumPy库。下面是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 构造数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.5, 3.7, 4.9, 6.1, 7.3])
# 绘制散点图
plt.scatter(x, y)
# 进行曲线拟合
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.plot(x, p(x), "r--")
# 添加图例和标签
plt.legend(["Fit line", "Scatter"], loc="upper left")
plt.title("Scatter plot with linear fit")
plt.xlabel("X")
plt.ylabel("Y")
# 显示图形
plt.show()
```
在这个例子中,我们使用polyfit函数进行线性拟合,然后使用poly1d函数将拟合结果转换为一个多项式对象。最后,我们使用plot函数将拟合的曲线绘制在散点图上。
阅读全文