Python中的plt.legend详解|极客教程

Python中的plt.legend详解

在Python中，我们经常会使用matplotlib库来绘制各种图形，而plt.legend()函数则用于在图形中添加图例。图例是用来标识不同数据系列的标签，并显示在图形中的特定位置。在本文中，我们将详细讨论plt.legend()函数的用法，包括图例的位置、样式、大小等属性。

plt.legend()函数介绍

plt.legend()函数用于在当前图形中添加图例。它的基本语法如下：

plt.legend(labels, loc='best', prop={'size': 10})

labels：图例标签的列表，用于标识不同数据系列。
loc：图例的位置，可以是字符串（如’upper right’、’lower left’等）或整数（0-10的数字，代表不同的位置）。默认值为’best’，表示自动选择最佳位置。
prop：图例字体的属性，可以设置字体大小、颜色等。

图例位置设置

图例的位置可以通过loc参数进行设置，常用的位置包括：

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

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], label='Series A')
plt.plot([4, 3, 2, 1], label='Series B')
plt.legend(loc='upper right')
plt.show()

图例样式设置

除了位置外，我们还可以通过prop参数设置图例的样式，比如字体大小、颜色等：

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], label='Series A')
plt.plot([4, 3, 2, 1], label='Series B')
plt.legend(prop={'size': 12, 'weight': 'bold', 'color': 'blue'})
plt.show()

多列图例设置

有时候图例较多，可以设置图例为多列显示，以免图例重叠，可以通过ncol参数设置多列显示的列数：

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], label='Series A')
plt.plot([4, 3, 2, 1], label='Series B')
plt.plot([2, 2, 2, 2], label='Series C')
plt.plot([3, 3, 3, 3], label='Series D')
plt.legend(ncol=2)
plt.show()

修改图例标签

如果需要对图例标签进行修改，可以在plt.legend()函数中传入labels参数，该参数为一个字符串列表，用于替换原有的图例标签：

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], label='Series A')
plt.plot([4, 3, 2, 1], label='Series B')
plt.legend(labels=['Data A', 'Data B'])
plt.show()

图例边框设置

默认情况下，图例是没有边框的，如果需要添加边框，可以通过frameon参数进行设置：

import matplotlib.pyplot as plt

plt.plot([1, 2, 3, 4], label='Series A')
plt.plot([4, 3, 2, 1], label='Series B')
plt.legend(frameon=True)
plt.show()