如何在Python中在散点图上叠加一条线？

在数据分析和可视化中，散点图经常被用来描绘两个变量之间的关系。然而，很多情况下我们需要在散点图上叠加一条线，用来更加清晰地表示变量之间的趋势。Python作为数据分析的常用工具之一，在matplotlib库的支持下，我们可以轻松地在散点图上绘制线条。

本文将着重讲解在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, label='Data points', color='blue')
plt.legend()
plt.show()

我们可以运行以上代码，生成一张简单的散点图，并准备在它上面叠加一条线。

在散点图上绘制折线

在matplotlib中，我们可以使用plot函数来绘制线条。具体而言，我们需要在数据上运用一定的插值方法，使得折线能够更好地贴合数据分布。

以下是代码示例，展示如何在散点图上绘制一条折线：

# 对数据进行排序
x_sorted = np.sort(x)
y_sorted = y[x.argsort()]

# 进行插值
f = np.polyfit(x_sorted, y_sorted, deg=1)   # 进行一次多项式拟合
y_pred = np.polyval(f, x_sorted)

# 绘制折线
plt.plot(x_sorted, y_pred, label='Fitted line', color='red')

plt.scatter(x, y, label='Data points', color='blue')
plt.legend()
plt.show()

我们将数据进行排序，并通过polyfit函数将数据进行一次多项式拟合。最后，我们使用plot函数在散点图上绘制一条折线。

修改线条颜色、宽度和样式

我们可以通过修改plot函数的参数，来对线条的样式进行进一步的设置。

以下是几个常用的参数：

color：线条颜色，取值可以为字符串（如'red'）或RGBA值（如(1, 0, 0, 0.5)）。
linewidth：线条宽度，取值为一个数字。
linestyle：线条样式，取值可以为字符串（如'--'、':'等）或者已经定义好的样式（如'dashed'、'dashdot'等）。

以下是代码示例，展示如何修改线条颜色、宽度和样式：

plt.plot(x_sorted, y_pred, label='Fitted line', color='red', linewidth=2, linestyle='--')

plt.scatter(x, y, label='Data points', color='blue')
plt.legend()
plt.show()