我正在为如下所示的数据集绘制散点图:

x = [1, 1, 2, 2, 3, 3, 4, 4]
y = [1, 2, 3, 4, 1, 2, 3, 4]
labels = [1, 3, 0, 2, 2, 1, 0, 3]

colors = np.array(plt.rcParams['axes.prop_cycle'].by_key()['color'])

plt.scatter(x, y, color=colors[labels])

如果我调用 plt.legend，对于整个数据集，将只显示一个条目，带有第一个符号。如何创建一个包含所有四个元素的图例，并显示为好像我绘制了四个单独的数据集？

可能相关:Matplotlib histogram with multiple legend entries
基于:Matplotlib, how to loop?

最佳答案

我认为总体上最简单的解决方案是将所有工作委托(delegate)给 matplotlib。此处描述了该方法:https://matplotlib.org/gallery/lines_bars_and_markers/scatter_with_legend.html#automated-legend-creation .对于这种简化的方法，使用 PathCollection's legend_elements方法就是您所需要的:

s = plt.scatter(x, y, c=labels)
plt.legend(*s.legend_elements())

更改颜色图或用其他内容(例如文本)替换标签是微不足道的:

text_labels = ['one', 'two', 'three', 'four']
s = plt.scatter(x, y, c=labels, cmap='jet', vmin=0, vmax=4)
plt.legend(s.legend_elements()[0], text_labels)

如果 labels 还不是 [0-n) 范围内元素的排序数组，可以使用 np.unique 轻松获得它:

labels = ['b', 'd', 'a', 'c', 'c', 'b', 'a', 'd']
text_labels, labels = np.unique(labels, return_inverse=True)

https://stackoverflow.com/questions/61905622/