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

标签 python matplotlib legend

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

我使用单个 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 组显示下一种颜色。 enter image description here

最佳答案

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()

enter image description here

关于python - 如何向 matplotlib 图例添加更多项?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22003924/

相关文章:

python - 文件名和路径的问题

python - Pandas DataFrame 中索引和列的级别数(深度)

python - 更改 Spark Web UI 的根路径?

python - 拒绝读取Python文件的部分内容

python - 如何使用 matplotlib 或 seaborn 在两行中绘制 5 个子图?

gnuplot:图例隐藏在数据后面

python - 在 Python 中创建阈值编码的 ROC 图

python - 带日期的 matplotlib

r - 如何在图例中获取 "type=b"行

Highcharts:按字幕结构图例