使用与matplotlib中的类标签对应的不同颜色绘制坐标

时间:2016-07-27 13:23:31

标签: python python-2.7 matplotlib

我有一个形状X的矩阵Nx2和一个相应的标签集(5个类的5个标签,例如,0,1,2,3,4),矩阵{{1长度Y。现在,我想使用matplotlib散点图绘制N,以使X中与X中每个标签对应的Y中的值具有不同的颜色。我不确定此问题的plt.scatter()的确切参数。

1 个答案:

答案 0 :(得分:4)

您可以将colors参数指定为scatter。如果只是将类标签数组作为颜色参数传递,则颜色图将用于将每个标签映射到相应的RGB值。

plt.scatter(X[:,0], X[:,1], c=Y)