Matplotlib:同一图形上的多个图例

Question

我有几个绘图函数，每个函数都接收不同类型的数据并用自己的标题和图例进行绘图。然而，当我试图在同一个图形上绘制它们时，前一个图形被覆盖。

我知道这样做的一种方法是调用get_artist（），但是由于图例是在函数内部创建的，所以这似乎是不可能的。是否有任何类型的LegendHandler可以执行此操作？从每个绘图中检索图例并在图形上显示它们？我应该从每个函数返回图例吗？

这是我所拥有的简化代码：

def plot_1(fig, data, ax = None):
    if ax = None:
        ax = fig.add_subplot(111)
    ax.plot(data)
    ax.set_xlabel('t')
    plt.axis('tight')

    # Create a legend
    colors = {'A':'blue',
              'B':'yellow',
              'C':'red'}
    legend_labels = []
    legend_handles = []

    for key in colors.keys():
        legend_labels.append(key)
        p = matplotlib.patches.Rectangle((0, 0), 1, 1, fc = colors[key], alpha = 0.5)
        legend_handles.append(p)
    ax.legend(legend_handles, legend_labels, loc='center left', bbox_to_anchor=(1, 0.5))

def plot_2(fig, data, ax = None):
    if ax = None:
        ax = fig.add_subplot(111)
    ax.plot(data, color='green', linewidth=1, label='L1')
    ax.set_xlabel('t')
    ax.autoscale()
    ax.legend()

def main():

    plot_1(fig, data = data1)
    plot_2(fig, data = data2)

我不确定这是否是最好的方法，但由于我多次使用这些绘图模型，我觉得我必须保持它们的函数形式。

• matplotlib:2个不同的图例在同一个图表上，这就是我提到的get\u artist，这是在该线程中提到的原因。我现在已经用它解决了这个问题，但我一直在为这个特殊的案例寻找一种更干净的方法。我现在所做的是从每个函数返回一个图例，并执行该线程中建议的操作

Answer 1

让你的功能回归传奇

def plot_2(fig, data, ax = None):
    ....
    return ax.legend()

那你就可以了

legend1 = plot_1(fig, data = data1)
plot_2(fig, data = data2)
pyplot.gca().add_artist(legend1)

这里建议：matplotlib：同一图形上的两个不同的图例