带有垂直线的matplotlib图例

在数据可视化中，图例起到了非常重要的作用。matplotlib作为Python中最常用的绘图库，其图例功能也相当强大。

但是在一些特殊的情况下，我们需要对图例进行进一步的修改，比如给图例添加垂直线。本文将介绍如何使用matplotlib中的Line2D方法在图例中添加垂直线。

示例

首先，我们需要用到的依赖库如下：

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

接下来，让我们来看一个例子。假设我们有如下的数据：

import numpy as np

x = np.linspace(-10, 10, 101)
y1 = x ** 2
y2 = x ** 3

我们希望绘制这两个曲线，并在图例中用一条垂直线将它们分开。代码如下：

fig, ax = plt.subplots()
line1 = ax.plot(x, y1, label='y1')
line2 = ax.plot(x, y2, label='y2')

# 添加垂直线到图例中
handles = [Line2D([0], [0], color='black', linewidth=1, linestyle='--'),
           line1[0],
           Line2D([0], [0], color='black', linewidth=1, linestyle='--'),
           line2[0]]
labels = ['sep', 'y1', 'sep', 'y2']
ax.legend(handles, labels)

plt.show()

可以看到，我们在图例中添加了一条垂直线，用于将两个曲线分开。在代码中，我们首先使用plot方法绘制了两个曲线，然后使用Line2D方法添加了一条黑色虚线，用于分割图例。最后，我们将Line2D和曲线对象传给legend方法，同时在labels中设置相应的标签，就可以生成我们想要的图例了。