我有一个形状X
的矩阵Nx2
和一个相应的标签集(5个类的5个标签,例如,0,1,2,3,4),矩阵{{1长度Y
。现在,我想使用matplotlib散点图绘制N
,以使X
中与X
中每个标签对应的Y
中的值具有不同的颜色。我不确定此问题的plt.scatter()
的确切参数。
答案 0 :(得分:4)
您可以将colors参数指定为scatter
。如果只是将类标签数组作为颜色参数传递,则颜色图将用于将每个标签映射到相应的RGB值。
plt.scatter(X[:,0], X[:,1], c=Y)