PyPlot：基於目標值的彩色散點圖數據

Question

我有一個二維數據框。 IE

data = [[  7.08535569   5.20423916]
 [ -6.12292297  -9.79831639]
 [ -0.14405156   0.02112219]
 ...
 [ -2.93838878   2.4744518 ]
 [  8.07134259 -10.8165695 ]
 [  5.14161567   2.68737039]]

我也有每個索引的目標值，即

target = [-1, 1, 1, -1 ... -1, 1, -1, -1]

使用 PyPlot 根據目標值對點進行顏色編碼的最簡單方法是什么？

目前我有

X = data[:,0]
y = data[:,1]

plt.plot(X, y, 'ro')
plt.show()

Answer 1

使用scatter ，它接受顏色圖的其他參數：

...

data = [[  7.08535569   5.20423916]
 [ -6.12292297  -9.79831639]
 [ -0.14405156   0.02112219]
 ...
 [ -2.93838878   2.4744518 ]
 [  8.07134259 -10.8165695 ]
 [  5.14161567   2.68737039]]

...

plt.scatter(x, y, c=target, s=500, cmap='gray')
plt.show()

...

參考：

Matplotlib 散點圖； 顏色作為第三個變量的函數