python画散点图如何添加趋势线

要在 Python 画散点图中添加趋势线，可以使用 NumPy 和 Matplotlib 库中的 polyfit() 和 plot() 函数。polyfit() 函数用于拟合数据点，plot() 函数用于绘制拟合线。

首先，导入 NumPy 和 Matplotlib 库，并创建一些随机数据点：

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(50)
y = np.random.rand(50)

然后，使用 polyfit() 函数拟合数据点，并计算拟合线的斜率和截距：

z = np.polyfit(x, y, 1)
p = np.poly1d(z)

这里的参数 1 表示拟合直线的阶数。如果您想拟合二次曲线或更高阶曲线，可以将参数设置为 2 或更高。

接下来，使用 plot() 函数绘制数据点和拟合线：

plt.scatter(x, y)
plt.plot(x, p(x), "r--")

这将在散点图上绘制数据点，并在数据点上绘制拟合线。

最后，使用 show() 函数显示图形：

plt.show()

完整代码如下：

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(50)
y = np.random.rand(50)

z = np.polyfit(x, y, 1)
p = np.poly1d(z)

plt.scatter(x, y)
plt.plot(x, p(x), "r--")

plt.show()

请注意，拟合直线的斜率和截距可能与您的数据点有关。因此，您需要根据需要调整拟合的阶数和拟合结果。