Python中的图例设置|极客笔记

Python中的图例设置

在Python中绘制图表时，添加图例是非常有必要的，因为它可以帮助读者更好地理解图表中各个数据系列的含义。在本文中，我们将详细介绍如何在Python中设置图例，并给出一些示例代码，帮助读者更好地掌握图例设置的技巧。

1. Matplotlib中的图例设置

Matplotlib是一个强大的绘图工具，在Matplotlib中，可以通过legend方法来添加图例。下面给出一个简单的示例代码：

import matplotlib.pyplot as plt

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

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

在上面的示例代码中，我们首先创建了两组数据y1和y2，然后分别绘制了两条折线，并通过label参数指定了每条折线对应的标签。最后调用legend方法添加图例，并调用show方法显示图表。

运行上面的代码，我们会得到一个包含图例的折线图。图例会自动根据每条折线的标签来设置。

2. 图例位置设置

在Matplotlib中，可以通过在legend方法中设置loc参数来指定图例的位置。常用的图例位置有以下几种：

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

下面是一个示例代码：

plt.legend(loc='upper right')

在上面的示例中，我们将图例位置设置为右上角。

3. 图例字体设置

在Matplotlib中，可以通过在legend方法中设置prop参数来指定图例的字体样式。常用的字体样式设置有以下几种：

size：字体大小
style：字体样式，如'normal'、'italic'、'oblique'
weight：字体粗细，如'normal'、'bold'、'light'
family：字体族，如'serif'、'sans-serif'、'cursive'、'fantasy'

下面是一个示例代码：

plt.legend(prop={'size': 12, 'weight': 'bold', 'family': 'sans-serif'})

在上面的示例中，我们将图例的字体大小设置为12，粗细设置为粗体，字体族设置为'sans-serif'。

4. 多个图例设置

在Matplotlib中，有时我们需要在一个图表中添加多个图例。这时，可以通过legend方法的handles参数来设置。下面是一个示例代码：

import matplotlib.lines as mlines

line1 = mlines.Line2D([], [], color='blue', marker='o', markersize=10, label='Line 1')
line2 = mlines.Line2D([], [], color='red', marker='s', markersize=10, label='Line 2')

plt.legend(handles=[line1, line2])
plt.show()

在上面的示例中，我们首先创建了两个Line2D对象line1和line2，然后通过handles参数将这两个对象传给legend方法，从而实现了在一个图表中添加多个图例的效果。