为fill_between（）段的不同颜色添加图例

Question

我正在创建一个＆＃34;事件情节＆＃34;目前看起来像这样：

但是，我不知道如何为每个颜色组添加图例。这就是当前情节的创建方式：

handles = dict()

for i, channel_events in enumerate(channel_event_list):

    for event in channel_events:
        start = event[0]
        end = event[1]
        y = (i, i + padding)

        c = 'red' if colors is None else colors[i]

        h = plt.fill_between([start, end], y[0], y2=y[1], color=c)

        if c not in handles:
            handles[c] = list()
        handles[c].append(h)

我认为我可以使用fill_between()的输出作为句柄，但似乎我错了。

那么，仅仅为这些颜色获取传奇的最简单方法是什么？

Answer 1

要通过fill_between电话或任何其他PolyCollection来创建图例处理，您可以使用此PolyCollection并将其提供给图例。

import matplotlib.pyplot as plt

h = plt.fill_between([1,2,3],[1,3,4], y2=[1,2,3], color="#c1009b")

plt.legend(handles=[h], labels=["MyLabel"])
plt.show()

更简单的方法是直接使用label图的fill_between参数（就像任何其他图一样）来创建自动图例条目。

import matplotlib.pyplot as plt

plt.fill_between([0,1,2],[4,3,4], y2=[3,2.5,3], color="#c1009b", label="Label1")
plt.fill_between([0,1,1.3],[1,2,0.5], y2=[0,-1,0], color="#005ec1", label="Label2")
plt.fill_between([2,3,4],[0.5,1,0], y2=[-1,-1,-1.5], color="#005ec1", label="_noLabel")

plt.legend()
plt.show()