matplotlib의 클래스 레이블에 해당하는 다른 색상으로 플롯 좌표 지정

Question

X 모양이 Nx2이고 해당 레이블 집합 (5 개 클래스에 대해 5 개의 레이블, 예 : 0,1,2,3,4)이 매트릭스에 있습니다. 길이가 N 인 Y입니다. 이제 X을 plot하여 X의 각 레이블에 해당하는 값이 Y이되도록 matplotlib scatter plot을 사용하고 싶습니다. 이 문제에 대한 plt.scatter()의 정확한 매개 변수에 대해 확실하지 않습니다.

Answer 1

색 매개 변수를 scatter으로 지정할 수 있습니다. 단순히 색상 매개 변수로 클래스 레이블 배열을 전달하면 색상 맵을 사용하여 각 레이블을 해당 RGB 값에 매핑합니다.

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