plt拟合散点
时间: 2023-07-28 22:06:52 浏览: 71
三维空间离散点的平面方程拟合
要使用Python中的matplotlib库中的plt.scatter()函数来绘制散点图,并使用numpy库中的polyfit()函数来进行拟合。下面是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成随机数据
x = np.random.rand(50)
y = 5*x + 0.2*np.random.randn(50)
# 绘制散点图
plt.scatter(x, y)
# 进行一次多项式拟合
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
# 绘制拟合曲线
plt.plot(x, p(x), "r-")
plt.show()
```
在这个例子中,我们生成了一些随机数据,并使用plt.scatter()函数绘制了散点图。然后,我们使用polyfit()函数进行一次多项式拟合,并使用poly1d()函数将拟合结果转换为一个方便计算的多项式对象。最后,我们使用plt.plot()函数绘制了拟合曲线,并使用"r-"参数指定曲线的颜色为红色。运行这段代码,你将会看到一个包含散点和拟合曲线的图形窗口。
阅读全文