matplotlib get_legend_handles_labels总是返回空

Question

当将元素添加到相同的图形中时，我遇到了问题。问题在于传说。在每次迭代时，我都会添加元素和相应的图例。但我希望这个传说包括所有迭代中的所有不同元素。

问题是函数get_legend_handles_labels()返回空句柄&标签列表。

守则如下：

    handles, labels = ax.get_legend_handles_labels()
    for key in legendMap.iterkeys():
        if key not in labels:
            handles.append(h[legendMap[key]])
            labels.append(key)

    ax.legend(handles, labels, loc='center left', bbox_to_anchor=(1, 0.5))
    h1,l1 = ax.get_legend_handles_labels()

h1、l1为空，而与句柄、标签相同。

下面的代码可以做到这一点，但我认为应该有更优雅的解决方案：

    legend = ax.get_legend()
    labels = [] if legend is None else [str(x._text) for x in legend.texts]
    handles = [] if legend is None else legend.legendHandles
    for key in legendMap.iterkeys():
        if key not in labels:
            handles.append(h[legendMap[key]])
            labels.append(key)

    ax.legend(handles, labels, loc='center left', bbox_to_anchor=(1, 0.5))

非常感谢您的帮助，谢谢！

Answer 1

添加曲线时，图表中的条形图指定标签，例如：

ax.plot(x, y, linewidth=1.75, color='#00ff00', label='curve1')

在那之后，一切都像魅力一样运作。