matplotlib 设置图例的颜色

Question

我正在使用 matplot lib 生成一个绘图，通过执行以下操作来绘制许多点（大约几千个）：

labels = []
for item in items:
        label = item[0]
        labels.append(label)
        plt.plot(item[1][0], item[1][1], 'ro', c = colors[item], label = str(label))

然后通过执行以下操作生成图例：

plt.legend([str(x) for x in np.unique(labels)])

但是，对于图例中的每个标签，对应的颜色都是相同的（不是图中的颜色）。有没有办法手动设置图例的颜色。

我附上了一个示例图来说明问题。

--编辑-- 只是按照某些人的建议调用plt.legend() 似乎并不能解决我的问题，它为每个点添加了一个图例条目。有关示例输出，请参见下图：

Answer 1

您还可以创建假线标记并将它们用作图例条目：

markers = [plt.Line2D([0,0],[0,0], color=color[item], marker='o', linestyle='') for item in np.unique(items)]
plt.legend(markers, np.unique(labels), numpoints=1)

你可以在this answer看到一个完整的例子。

请注意，我尚未对此进行测试，您可能需要将 np.unique(items) 调整为您的实际数据集。 color 被假定为为您的项目提供颜色的字典。

Answer 2

这应该可行：

labels = []
for item in items:
    label = item[0]
    plt.plot(item[1][0], item[1][1], 'o', c=colors[item], label=str(label))
plt.legend()

如果您在创建艺术家时直接指定标签（在对plot 的调用中），您可以只调用plt.legend() 而无需参数。它将遍历当前轴上的艺术家并使用他们的标签。这样图例中的颜色将与图中的颜色相匹配。