python画拟合曲线和散点图
时间: 2023-10-06 10:14:52 浏览: 226
利用python绘制散点图
4星 · 用户满意度95%
在中,您可以使用matplotlib库来画拟合曲线和散点图。首先,您需要导入matplotlib库,并使用它的pyplot模块来创建图表。然后,使用plot函数来绘制散点图,使用polyfit函数计算回归系数,并使用poly1d函数创建拟合曲线。最后,使用scatter函数来绘制散点图和拟合曲线。
下面是一个示例代码:
```
import matplotlib.pyplot as plt
import numpy as np
# 创建数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 4, 5, 6])
# 绘制散点图
plt.scatter(x, y)
# 计算回归系数
coefficients = np.polyfit(x, y, 1)
polynomial = np.poly1d(coefficients)
# 绘制拟合曲线
plt.plot(x, polynomial(x))
# 显示图表
plt.show()
```
阅读全文