Python legend位置详解

介绍

在使用Python绘制图表时，我们经常需要添加图例(legend)，以便于解释图表中每个数据系列对应的含义。图例可以放置在图表的各个位置，以符合我们的展示要求。本文将详细介绍Python中图例的各种位置设置。

准备工作

在开始之前，我们需要安装matplotlib库，它是Python的一个常用绘图库，并且提供了丰富的配置选项。

!pip install matplotlib

安装完成后，我们就可以开始学习如何设置图例的位置了。

一、图例的基本设置

在绘制图表之前，我们首先需要创建一些示例数据。在本文中，我们使用一个简单的折线图作为示例。

import matplotlib.pyplot as plt

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

plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
plt.legend()
plt.show()

上述代码中，我们定义了两个数据系列y1和y2，并绘制了折线图。plt.plot()函数用于绘制折线图，其中的label参数用于设置数据系列的名称。plt.legend()函数用于显示图例。

运行以上代码，可以看到图例默认显示在图表的右上角。

二、图例的位置属性

图例的位置属性可以通过loc参数来设置。loc参数接受一个整数或一个字符串作为输入，用于指定图例显示的位置。

plt.legend(loc='upper right')

其中常用的位置属性包括：

‘best’：自动选择最佳位置
‘upper right’：右上角
‘upper left’：左上角
‘lower right’：右下角
‘lower left’：左下角
‘center right’：右侧中央
‘center left’：左侧中央
‘center’：图表正中心

示例代码如下：

plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
plt.legend(loc='upper right')
plt.show()

三、图例的边框设置

通过frameon参数，我们可以控制图例的边框是否显示。

plt.legend(frameon=False)

设置frameon=False，可以隐藏图例的边框。

四、图例的背景透明度设置

通过facecolor参数，我们可以设置图例的背景颜色。同时，通过alpha参数，我们还可以设置图例的背景透明度。

plt.legend(facecolor='gray', alpha=0.5)

设置facecolor='gray'，可以将图例的背景颜色设置为灰色；设置alpha=0.5，可以将图例的背景透明度设置为50%。

五、图例的大小和间距设置

通过prop参数，我们可以设置图例文本的大小。同时，通过bbox_to_anchor参数，我们还可以设置图例的位置。

plt.legend(prop={'size': 12}, bbox_to_anchor=(1, 0.5))

设置prop={'size': 12}，可以将图例文本的大小设置为12；设置bbox_to_anchor=(1, 0.5)，可以将图例放置在图表的右侧中央。

六、多个图例的处理

有时候，我们可能需要在一个图表中显示多个图例。在matplotlib中，可以通过创建多个Legend对象来实现。

line1, = plt.plot(x, y1, label='Line 1')
line2, = plt.plot(x, y2, label='Line 2')
legend1 = plt.legend(handles=[line1], loc='upper left')
legend2 = plt.legend(handles=[line2], loc='lower right')
plt.gca().add_artist(legend1)
plt.gca().add_artist(legend2)
plt.show()

在上述代码中，我们首先创建了两个折线图，并分别为它们创建了单独的图例。接着，使用plt.gca().add_artist()函数将两个图例添加到图表中。