我有兴趣在我的散点图中绘制一个图例。我当前的代码是这样的
x=[1,2,3,4]
y=[5,6,7,8]
classes = [2,4,4,2]
plt.scatter(x, y, c=classes, label=classes)
plt.legend()
问题是在创建绘图时，图例显示为数组，而不是显示唯一标签及其类。
我知道这是之前在one 等线程中讨论过的一个问题，但是我觉得我的问题更简单，并且那里的解决方案不适合它。此外，在该示例中，该人正在指定颜色，但在我的情况下，我事先知道我需要多少种颜色。此外，在this 示例中，用户正在创建多个散点图，每个散点图具有唯一的颜色。同样，这不是我想要的。我的目标是使用 x,y 数组和标签简单地创建绘图。这可能吗？
谢谢。
【问题讨论】：
标签： python matplotlib plot
实际上，这两个链接问题都提供了一种实现预期结果的方法。
最简单的方法是创建与存在的唯一类一样多的散点图，并为每个类指定一个颜色和图例条目。
import matplotlib.pyplot as plt
x=[1,2,3,4]
y=[5,6,7,8]
classes = [2,4,4,2]
unique = list(set(classes))
colors = [plt.cm.jet(float(i)/max(unique)) for i in unique]
for i, u in enumerate(unique):
xi = [x[j] for j in range(len(x)) if classes[j] == u]
yi = [y[j] for j in range(len(x)) if classes[j] == u]
plt.scatter(xi, yi, c=colors[i], label=str(u))
plt.legend()
plt.show()
如果类是字符串标签，解决方案看起来会略有不同，因为您需要从它们的索引中获取颜色，而不是使用类本身。
import numpy as np
import matplotlib.pyplot as plt
x=[1,2,3,4]
y=[5,6,7,8]
classes = ['X','Y','Z','X']
unique = np.unique(classes)
colors = [plt.cm.jet(i/float(len(unique)-1)) for i in range(len(unique))]
for i, u in enumerate(unique):
xi = [x[j] for j in range(len(x)) if classes[j] == u]
yi = [y[j] for j in range(len(x)) if classes[j] == u]
plt.scatter(xi, yi, c=colors[i], label=str(u))
plt.legend()
plt.show()
【讨论】：
scatterlegend(sc, values, classes=None) 函数。如果您需要帮助，何不提出一个新问题。
在这里手动填写table 可能会很有用。如果你的类是连续的数字，另一个想法是使用colorbar。我将两种方法合二为一。
import matplotlib.pyplot as plt
import numpy as np
x=[1,2,3,4,5,6,7]
y=[1,2,3,4,5,6,7]
classes = [2,4,4,2,1,3,5]
cmap = plt.cm.get_cmap("viridis",5)
plt.scatter(x, y, c=classes, label=classes,cmap=cmap,vmin=0.5,vmax=5.5)
plt.colorbar()
unique_classes = list(set(classes))
plt.table(cellText=[[x] for x in unique_classes], loc='lower right',
colWidths=[0.2],rowColours=cmap(np.array(unique_classes)-1),
rowLabels=['label%d'%x for x in unique_classes],
colLabels=['classes'])
【讨论】：