matplotlib图例|极客笔记

matplotlib图例

在使用matplotlib绘制图形时，图例是非常重要的。图例能够帮助我们理解图中的数据点对应的含义，从而更好地解读图形。本文将详细介绍如何在Matplotlib中使用图例，并展示一些常见的图例样式。

添加图例

在Matplotlib中，我们可以通过调用plt.legend()函数来添加图例。在调用此函数之前，我们需要在绘制图形时为每个数据集指定一个label，这样图例才能知道如何标识数据。

下面是一个简单的示例代码，演示如何在Matplotlib中添加图例：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend()
plt.show()

在上面的示例中，我们为两个数据集y1和y2分别添加了’label’参数，然后调用plt.legend()函数来添加图例。运行此代码后，就会在图中看到包含’sin(x)’和’cos(x)’的图例。

图例位置

默认情况下，Matplotlib会将图例放在合适的位置，以不遮挡数据点。但有时候我们可能希望将图例放在特定位置，可以通过传入loc参数来设置图例的位置。

下面是一个示例代码，演示如何将图例放置在右上角：

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend(loc='upper right')
plt.show()

在上面的代码中，我们传入了loc='upper right'参数，将图例放置在右上角。

图例样式

除了设置图例的位置，我们还可以设置图例的样式，比如图例的背景颜色、边框、文字颜色等。下面是一个示例代码，演示如何设置图例样式：

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend(loc='upper right', fontsize=12, fancybox=True, shadow=True, facecolor='lightblue')
plt.show()

在上面的代码中，我们设置了图例的字体大小为12，背景为浅蓝色，开启了阴影效果。

图例标题

有时候我们可能希望给整个图例添加一个标题，以便更好地说明数据的含义。在Matplotlib中，我们可以通过title参数来添加图例标题。

下面是一个示例代码，演示如何为图例添加标题：

plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend(title='Trigonometric Functions')
plt.show()

在上面的代码中，我们传入了title='Trigonometric Functions'参数，为图例添加了一个标题。

自定义图例

除了使用Matplotlib提供的默认图例外，我们还可以自定义图例，比如使用不同形状或颜色的标记来代表不同的数据集。下面是一个示例代码，演示如何自定义图例：

plt.plot(x, y1, label='sin(x)', color='blue', marker='o')
plt.plot(x, y2, label='cos(x)', color='red', marker='s')
plt.legend()
plt.show()

在上面的代码中，我们为两个数据集指定了不同的颜色和标记，并通过label参数指定了图例标题。

结论

本文介绍了在Matplotlib中添加图例的基本方法，包括如何添加图例、设置图例位置、样式、标题以及自定义图例。通过合理地使用图例，我们可以更好地理解图中的数据，并进行更准确的分析和解读。

matplotlib图例