matplotlib（python）-为多个图创建一个自定义图例，而无需pyplot

Question

我想在pyqt GUI中为matplotlib（python）中的多个绘图创建一个自定义图例。（pyqt建议不要使用pyplot，因此必须使用面向对象的方法）。

网格中将显示多个绘图，但用户可以定义要显示的绘图数量。我希望图例显示在所有绘图的右侧，因此我不能简单地为最后绘制的轴创建图例。我希望为整个图形创建图例，而不仅仅是最后一个轴（类似于pyplot中的plt.figlegend）。

在我在别处看到的示例中，这需要引用绘制的线。同样，我不能这样做，因为用户可以选择在图形上显示哪些线，我希望图例总是显示所有可能的线，无论它们当前是否显示。

（注意下面的示例代码使用pyplot，但我的最终版本不能）

import matplotlib.pyplot as plt
import matplotlib.lines as mlines
import numpy as np

fig = plt.figure()

# Create plots in 2x2 grid
for plot in range(4):
    # Create plots
    x = np.arange(0, 10, 0.1)
    y = np.random.randn(len(x))
    y2 = np.random.randn(len(x))
    ax = fig.add_subplot(2,2,plot+1)
    plt.plot(x, y, label="y")
    plt.plot(x, y2, label="y2")

# Create custom legend
blue_line = mlines.Line2D([], [], color='blue',markersize=15, label='Blue line')
green_line = mlines.Line2D([], [], color='green', markersize=15, label='Green line')
ax.legend(handles=[blue_line,green_line],bbox_to_anchor=(1.05, 0),  loc='lower left', borderaxespad=0.)

如果我换斧头。图例到：fig.legend（句柄=[蓝色线，绿色线]），然后python生成错误：

TypeError:legend（）至少接受3个参数（给定2个）

（我猜是因为直线点没有被引用）

谢谢你提供的任何帮助-我已经看了一个星期了！

Answer 1

您得到的错误是因为图。图例要求您同时传递句柄和标签。

从文档中：

图例（手柄、标签、*args、**kwargs）

在图中放置图例。标签是一个字符串序列，句柄是一个Line2D或Patch实例序列。

以下作品：

# Create custom legend
blue_line = mlines.Line2D([], [], color='blue',markersize=15, label='Blue line')
green_line = mlines.Line2D([], [], color='green', markersize=15, label='Green line')

handles = [blue_line,green_line]
labels = [h.get_label() for h in handles] 

fig.legend(handles=handles, labels=labels)