用法介绍Python Legend函数|极客教程

用法介绍Python Legend函数

在Python的matplotlib库中，legend函数用于在绘图中添加图例。图例是指示每条曲线或每个散点的标签，以帮助读者理解图表中的数据。

legend函数的基本用法

legend函数可以接受一个包含图例标签的列表作为参数，并将图例添加到当前图形中。下面是legend函数的基本语法：

import matplotlib.pyplot as plt

# 创建图形
plt.plot([1, 2, 3, 4], label='Line 1')
plt.plot([4, 3, 2, 1], label='Line 2')

# 添加图例
plt.legend()

# 显示图形
plt.show()

在这个示例中，我们使用plot函数绘制了两条线，并在每条线上添加了一个标签。然后通过调用legend函数，将这两个标签添加到图例中。最后调用show方法显示出图形。

图例位置和样式设置

legend函数还可以接受一些关键字参数来控制图例的位置、字体和样式。下面是一些常用的参数和取值：

loc：控制图例的位置，可以取值为’best’, ‘upper right’, ‘upper left’, ‘lower left’, ‘lower right’, ‘right’, ‘center left’, ‘center right’, ‘lower center’, ‘upper center’, ‘center’。
fontsize：控制图例中文本的字体大小。
ncol：控制图例的列数。
title：设置图例的标题。

import matplotlib.pyplot as plt

# 创建图形
plt.plot([1, 2, 3, 4], label='Line 1')
plt.plot([4, 3, 2, 1], label='Line 2')

# 添加图例
plt.legend(loc='upper right', fontsize='large', ncol=2, title='Legend')

# 显示图形
plt.show()

在这个示例中，我们设置图例在右上角显示，字体大小为large，分成两列显示，并设置了标题为’Legend’。

图例标签格式设置

在图例中，我们还可以设置标签的格式，比如修改颜色、样式和粗细等。legend函数中可以通过传递labelcolor、labelspacing、handlelength、handleheight等参数来进行设置。

import matplotlib.pyplot as plt

# 创建图形
plt.plot([1, 2, 3, 4], label='Line 1', color='red', linestyle='--', linewidth=2)
plt.plot([4, 3, 2, 1], label='Line 2', color='blue', linestyle=':', linewidth=2)

# 添加图例
plt.legend(labelcolor='black', labelspacing=1, handleheight=2)

# 显示图形
plt.show()

在这个示例中，我们设置了两条线的颜色、线型和线宽，然后使用legend函数将标签的颜色改为黑色，标签间的空白距离为1，图例标识高度为2。

使用handles和labels参数自定义图例

有时候我们需要自定义图例的内容和位置，可以使用handles和labels参数。handles参数用于传递一个包含图例标识的列表，labels参数用于传递与每个标识相对应的标签。

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

# 创建图形
plt.plot([1, 2, 3, 4], color='red')
plt.plot([4, 3, 2, 1], color='blue')

# 定义图例标识和标签
line1 = Line2D([0], [0], color='red', linewidth=2)
line2 = Line2D([0], [0], color='blue', linewidth=2)
handles = [line1, line2]
labels = ['Line 1', 'Line 2']

# 添加自定义图例
plt.legend(handles=handles, labels=labels, loc='upper right')

# 显示图形
plt.show()

在这个示例中，我们使用Line2D类创建了两个标识，分别代表红色线和蓝色线，然后传递给handles参数。labels参数对应每个标识的标签。最后通过legend函数将自定义图例添加到图形中。