按传说排队

Question

当我知道标签时，有没有方便的方法来获取线条的颜色？

现在我正在做

    color = None
    for l in ax.lines:
        if l.get_label() == myLabel:
            color = l.get_color()
            break

    else:
        raise Exception('not found')

并不像pythonic。

Answer 1

略好一点，但看起来还不是很好：使用传说标签列表：

ax.lines[ax.get_legend_handles_labels()[1].index(myLabel)].get_color()

Answer 2

您可以使用next：

color = next((l.get_color() for l in ax.lines if l.get_label() == myLabel), None)