如何使用Matplotlib将图例添加到散点图

在Python中，Matplotlib是一个用于绘制图形和图表的强大库。你可以使用它来创建各种类型的图表，包括散点图。如果你想在散点图中添加图例以标识不同的数据系列，你可以使用以下步骤：

1. 导入所需的库：

   import matplotlib.pyplot as plt

2. 准备数据：
   你需要准备两个数据集：一个是x轴的数据，另一个是y轴的数据。此外，你还需要一个颜色列表，用于为每个数据系列指定颜色。

   x = [1, 2, 3, 4, 5]
   y = [2, 3, 5, 7, 11]
   colors = ['red', 'green', 'blue', 'yellow', 'purple']

3. 创建散点图：
   使用Matplotlib的scatter()函数创建散点图。你需要指定x轴和y轴的数据，并为每个数据点指定一个颜色。为了实现这一点，你可以使用一个循环来迭代数据和颜色列表，并为每个数据点指定一个颜色。

   fig, ax = plt.subplots()
   for i in range(len(x)):
   ax.scatter(x[i], y[i], color=colors[i])

4. 添加图例：
   为了添加图例，你可以使用legend()函数。你可以通过将图例标签与每个颜色关联起来，来标识不同的数据系列。

   ax.legend(['Data series 1', 'Data series 2', 'Data series 3', 'Data series 4', 'Data series 5'])

5. 显示图表：
   最后，使用show()函数显示图表。

   plt.show()