python散点图拟合直线

要在 Python 中绘制散点图并拟合直线，可以使用 NumPy 和 Matplotlib 库。以下是一些步骤：

1. 导入库：

import numpy as np
import matplotlib.pyplot as plt

2. 创建数据：

x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 4, 5, 6])

3. 绘制散点图：

plt.scatter(x, y)

4. 使用 `numpy.polyfit()` 函数拟合直线：

slope, intercept = np.polyfit(x, y, 1)

其中，参数 `1` 表示拟合一次多项式。`polyfit()` 函数返回一个包含斜率和截距的数组。

5. 绘制拟合直线：

plt.plot(x, slope * x + intercept)

6. 显示图像：

plt.show()

完整代码如下：

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])

plt.scatter(x, y)

slope, intercept = np.polyfit(x, y, 1)

plt.plot(x, slope * x + intercept)

plt.show()

运行代码后，应该会看到一张包含散点图和拟合直线的图像。