python – 为散点图添加图例

这个问题已在SO上提出,但我想找到一个更清晰的解决方案.

给定X是100×2数据,标签是标签向量(从1到9)我绘制散点图如下：

pl.scatter(X[:,0],X[:,1],c = labels)
pl.show()

如何添加图例来解释一行代码中的颜色？其他解决方案分别绘制每个标签

a = pl.scatter(X1[:,X1[:,color = "red")
b = pl.scatter(X2[:,X2[:,color = "green")
c = pl.scatter(X3[:,X3[:,color = "blue")
pl.legend((a,b,c),("line 1","line 2","line 3")
pl.show()

解决方法

pl.scatter(X[:,c = labels)

for item in labels:
    #dummy plot just to create the legend
    pl.scatter([],[],c = item,label = item)
#loc = 0 is for the best position of the legend
#scatterpoints = 1 will only show one point in the legend instead of multiple points
plt.legend(loc = 0,scatterpoints = 1)
pl.show()

（编辑：安卓应用网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!