Matplotlib 多颜色图例项

Matplotlib是Python中广受欢迎的可视化工具之一，它支持多种类型的图表，并提供了许多可定制化的选项，其中包括图例项。在Matplotlib中，图例项用来解释每个线条、颜色图或其他元素的含义。本文将介绍如何在Matplotlib中创建多颜色图例项。

阅读更多：Matplotlib 教程

示例

在Matplotlib中创建多颜色图例项非常简单。我们可以在plt.legend()函数中使用多个handles和labels来创建多个图例项。例如，我们可以使用以下代码创建一个有三个颜色图例项的折线图：

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 2, 3, 4, 5]
y2 = [2, 3, 4, 5, 6]
y3 = [3, 4, 5, 6, 7]

plt.plot(x, y1, color='red', label='Line 1')
plt.plot(x, y2, color='blue', label='Line 2')
plt.plot(x, y3, color='green', label='Line 3')

plt.legend(handles=[plt.Line2D([], [], color='red', label='Line 1'),
                     plt.Line2D([], [], color='blue', label='Line 2'),
                     plt.Line2D([], [], color='green', label='Line 3')])

plt.show()

在上述代码中，我们使用三个不同颜色的折线绘制了一副简单的折线图，并使用了handles和labels参数来创建三个颜色的图例项。注意，我们使用了plt.Line2D()函数来创建每个图例项的线条，而不是使用每个数据集的线条。

多颜色散点图

除了折线图，我们也可以使用相同的的方法来创建多颜色散点图。例如，以下代码创建一个有多个颜色散点图的图表：

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(30)
y = np.random.rand(30)
colors = np.random.rand(30)

plt.scatter(x, y, c=colors)

plt.colorbar()

plt.legend(handles=[plt.Line2D([], [], marker='o', color='white', markerfacecolor='red', label='Red'),
                     plt.Line2D([], [], marker='o', color='white', markerfacecolor='blue', label='Blue'),
                     plt.Line2D([], [], marker='o', color='white', markerfacecolor='green', label='Green')])

plt.show()

在上述代码中，我们创建了一个有多个颜色散点图的图表。注意，我们使用c参数指定了每个散点的颜色，使用plt.colorbar()函数为图表添加了一个颜色条，并使用plt.Line2D()函数来创建每个图例项的点标记以及使用markerfacecolor参数设置点标记的颜色。