帮我用Python写一段代码,画一个线性回归图,蓝色的线,周边分布一些不在线上的黄色点
时间: 2024-02-06 16:11:29 浏览: 23
好的,以下是用Python画一个线性回归图,蓝色的线,周边分布一些不在线上的黄色点的代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成一些随机数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 4, 5, 6])
# 计算斜率和截距
slope, intercept = np.polyfit(x, y, 1)
# 用斜率和截距创建一个线性方程
line = slope * x + intercept
# 绘制原始数据和线性回归线
plt.scatter(x, y, c='yellow')
plt.plot(x, line, c='blue')
# 绘制一些随机的不在线上的数据点
for i in range(10):
x_new = np.random.randint(1, 6)
y_new = np.random.randint(2, 7)
plt.scatter(x_new, y_new, c='yellow')
# 显示图表
plt.show()
```
这段代码将生成一个线性回归图,其中包括原始数据点和线性回归线,以及一些随机的黄色点,这些点不在线性回归线上。你可以通过修改x和y数组来使用自己的数据,也可以修改颜色和点的数量来满足你的需求。