matplotlib中對應於類別標籤的不同顏色的繪圖座標

Question

我有一個形狀Nx2的矩陣X和矩陣的相應標籤組（5個標籤，例如，0,1,2,3,4） Y，長度爲N。現在我想繪製X使用matplotlib散點圖，使Y中的每個標籤對應的X中的值以不同的顏色顯示。我不確定這個問題的確切參數plt.scatter()。

Answer 1

您可以指定scatter的顏色參數。如果您只是將類標籤數組作爲顏色參數傳遞，則會使用顏色表將每個標籤映射到相應的RGB值。

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