Цветная диаграмма рассеяния по вектору

Question

У меня есть два вектора:

X=np.array(np.matrixlib.defmatrix.matrix('[3,3,3,8,5,2,4,6,3,8,2,0,4,1,8,9,9,8,7,2,2,9,8,7,1,2,7,7,4,3,2,6,9;7,5,1,5,3,9,5,7,10,7,8,2,3,1,6,7,8,8,3,8,8,9,4,9,4,7,5,0,8,6,10,2,9]'))


labels=np.array(np.matrixlib.defmatrix.matrix('[1,1,3,2,3,4,1,4,4,2,1,1,3,1,2,2,2,2,3,1,1,2,2,4,1,1,2,3,4,1,4,3,2]'))


#plot the data
plt.plot(X.T[:,0],X.T[:,1],'.'); 
plt.scatter(X.T[:,0],X.T[:,1],s=50,marker='o',c="b")
plt.title('Plot of our points');

Я хотел бы создать диаграмму рассеяния, но раскрасить ее в соответствии с вектором labels , Как я могу это сделать?

Answer 1

Понял, проблема была с чтением моего вектора меток.

labels=np.array(np.matrixlib.defmatrix.matrix('[1,1,3,2,3,4,1,4,4,2,1,1,3,1,2,2,2,2,3,1,1,2,2,4,1,1,2,3,4,1,4,3,2]'))

Я изменил это на

labels=np.array([1,1,3,2,3,4,1,4,4,2,1,1,3,1,2,2,2,2,3,1,1,2,2,4,1,1,2,3,4,1,4,3,2])

, и это в основном решает проблему.

Answer 2

Заменить ключевое слово: c=labels. См. разброс для получения дополнительной информации