python散点图中绘制直线代码

在 Python 中，可以使用 matplotlib 库来绘制散点图和直线。以下是一个示例代码，展示如何在散点图中绘制一条直线：

import matplotlib.pyplot as plt
import numpy as np

# 生成散点数据
x = np.random.rand(50)
y = np.random.rand(50)

# 绘制散点图
plt.scatter(x, y)

# 定义直线的斜率和截距
k = 0.5
b = 0.2

# 计算直线上的点
x_line = np.linspace(0, 1, 100)
y_line = k * x_line + b

# 绘制直线
plt.plot(x_line, y_line, color='red')

# 显示图形
plt.show()

该代码首先生成了一些随机的散点数据，并使用 `plt.scatter()` 函数绘制了散点图。然后，定义了直线的斜率和截距，并使用 `np.linspace()` 函数生成了直线上的点。最后，使用 `plt.plot()` 函数绘制了直线，并指定颜色为红色。调用 `plt.show()` 函数可以显示图形。