Matplotlib 保存图像时将图例放置在图像外面

Matplotlib是一个Python数据可视化库，它可以让用户创建各种类型的图表，包括折线图、散点图、柱状图和饼图等。在创建一个图表时，通常需要添加图例以解释数据，然而，在保存图像时，图例可能会遮挡数据。本文将介绍如何将图例放在图像外部以解决这一问题。

阅读更多：Matplotlib 教程

添加图例

在Matplotlib中，图例是用于向读者解释图表数据内容的重要部分，因此，添加图例是创建图表时必不可少的一步。下面是一个简单的示例，演示如何添加图例：

import matplotlib.pyplot as plt

x = [1, 2, 3, 4]
y = [1, 4, 9, 16]

plt.plot(x, y, label='Data')
plt.legend()
plt.show()

在上面的示例中，我们使用 legend 方法添加了图例。该方法默认将图例放在绘图区域内部的最佳位置。

对于包含很多数据的复杂图表，可能需要更多的空间来容纳图例，否则它可能会遮挡数据。此时，将图例放置在图像外部可能是一个更好的选择。下面将介绍如何在保存图像时将图例放置在图像外部。

保存图像及其图例

在Matplotlib中，可以使用 savefig 方法将图像保存为文件。该方法可以接受一个文件名并保存图像，但它只能保存绘图区域内的内容。为了将图例放置在图像外部，我们需要调整图像的大小和位置以容纳图例。

下面是一个具体的示例。假设我们有一张包含两个数据集的图表，并希望将它保存到一个名为 example.png 的文件中，并将图例放置在图像的右侧。

import matplotlib.pyplot as plt

# 创建示例数据
x1 = [1, 2, 3, 4]
y1 = [1, 5, 3, 8]
x2 = [1, 2, 3, 4]
y2 = [2, 3, 5, 1]

# 创建图表
fig, ax = plt.subplots()

# 绘制数据
ax.plot(x1, y1, label='Data 1')
ax.plot(x2, y2, label='Data 2')

# 添加图例
legend = ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))

# 调整图像大小和位置
fig.savefig('example.png', bbox_extra_artists=(legend,), bbox_inches='tight')

在上面的示例代码中，我们使用 bbox_extra_artists 参数指定了一个额外的艺术家列表，即图例对象。这将确保图例被包含在保存的图像中。我们还使用 bbox_inches 参数指定图像的边界框大小，以确保图例被正确地放置在图像外部。

运行上面的代码后，你将会得到一个名为 example.png 的文件，其中包含一个具有右侧图例的图表。

通过将图例放在图像的外部，我们可以获得更多的空间来展示数据，同时也不必担心图例会遮挡任何数据。可以在选择图例位置时尝试不同的参数，并根据需要调整图像的大小和位置。