python - 如何向 matplotlib 图例添加更多项？

Question

当我几个小时前遇到这个问题时，我认为只需要几分钟的谷歌搜索就可以解决，但我就是这样。

我使用单个 plt.bar() 调用绘制所有值:

plt.bar(range(len(vals)), vals, width=1, color=colors)

这将创建一个条形图，其中的条形采用颜色编码。当我尝试创建图例时，它只显示一种颜色和一个标签。因此，例如这样做:

labels = ['Group 1', 'Group 2', 'Group 3']
plt.legend(labels,
   loc='best',
   ncol=3)

仅在图例上显示“第 1 组”以及多种条形颜色之一。如何向图例添加更多项目？

我知道这个问题是因为我在一个图中分配了多种颜色，但在这种情况下，这是该部分最干净的解决方案。

我发现建议使用“ax.get_legend_handles_labels()”的资源，但这会为我返回两个空列表。

编辑:

@falsetru,这是一个代码示例，它描述了应用您的解决方案后的问题。我包括了所有相关部分。

import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np

items= ['item1','item2','item3']           #item names
vals = [1, 2, 3]                           #values to be plotted
groups = [['item1','item2'], ['item3']]    #defines item groupings

colors = [''] * len(items)
color_space = iter(cm.Paired(np.linspace(0, 1, len(groups))))
for c in groups:
    col = next(color_space)
    for l in c:
        i = items.index(l)
        colors[i] = col

for i, (val, c) in enumerate(zip(vals, colors)):
    plt.bar([i], [val], width=1, color=c)

labels = ['Group 1', 'Group 2']
plt.legend(labels, loc='best')

plt.show()

这是输出，但我希望第 2 组显示下一种颜色。

Answer 1

将vals(y 值)、indices(x 值)分组，就像对groups 所做的那样:

import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np

indices = [[1, 3], [2], [4]]               # <-----
vals = [[1, 2.5], [1.5], [3.5]]            # <-----
labels = ['Group 1', 'Group 2', 'Group 3']

color_space = iter(cm.Paired(np.linspace(0, 1, len(vals))))
for xs, ys, label in zip(indices, vals, labels):
    col = next(color_space)
    plt.bar(xs, ys, width=1, color=col, label=label)

plt.legend(loc='best')
plt.show()