python绘制拟合曲线和散点图
时间: 2023-10-19 20:07:15 浏览: 174
利用python绘制散点图
4星 · 用户满意度95%
您可以使用matplotlib库来绘制拟合曲线和散点图。具体步骤如下:
1. 导入matplotlib库和numpy库
```python
import matplotlib.pyplot as plt
import numpy as np
```
2. 准备数据
假设有一组数据,分别为x和y,可以使用numpy库的polyfit函数拟合出一条曲线。
```python
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 5, 6, 8])
z = np.polyfit(x, y, 1) # 拟合出一次函数
p = np.poly1d(z) # 生成拟合函数
```
3. 绘制散点图和拟合曲线
```python
plt.scatter(x, y) # 绘制散点图
plt.plot(x, p(x), "r--") # 绘制拟合曲线
plt.show() # 显示图像
```
完整代码如下:
```python
import matplotlib.pyplot as plt
import numpy as np
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 5, 6, 8])
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.scatter(x, y)
plt.plot(x, p(x), "r--")
plt.show()
```
阅读全文