Плотные кластеры с цветом из иерархической кластеризации

Question

#Plot the figure

plt.figure(1);
plt.clf();
plt.scatter(data[0,:],data[1,:]);
plt.xlabel('x');
plt.ylabel('y');
plt.title("Plot of all points from CURE dataset")

#Make dendrogram 

Y=pdist(data,'euclid');
distances=squareform(Y);
c=linkage(data,method='single');
dendrogram(Z);
idx = fcluster(Z,5,'maxclust');

Я хочу раскрасить точки в соответствии с тем, в каком кластере они принадлежать, как я могу это сделать?

Answer 1

Хорошо, я разобрался с ответом. Просто возьмите значения idx, которые представляют кластеры, и построите график с помощью этой команды:

plt.scatter(data[:,0], data[:,1], c=idx, cmap='rainbow')