Discussion :

[colapili]

Bonjour,

Je rencontre quelques problème avec les CMAPs.

En premier lieu j'affiche une heatmap à partir d'une liste de clusters issue d'un Kmeans. Par exemple avec trois cluster et un vecteur de 10 j'ai (1,1,1,1,2,2,2,1,3,3) .
J'ai donc une image qui a trois couleurs.

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
Number_of_cluster = 3
 
KMEANS = KMeans(Number_of_cluster, random_state=200)
cluster_labels = KMEANS.fit_predict(EMSC_corr2)
 
cluster_labels = pd.DataFrame(cluster_labels, index = EMSC_corr2.index )
 
# Tableau avec N°cluster et bons indices
a = np.empty((len(DATA_ORIGINA)))
a[:] = np.nan
a = pd.DataFrame(a)
a.loc[GOOD_SPECTRUM['DATA_ORIGINA'].index,:]= cluster_labels
 
fig = plt.figure(1, figsize=(6, 6))
XXX = np.reshape(a.values,(NbX, NbY))
plt.imshow(XXX, cmap='tab20c')

Cependant je souhaiterais afficher les centroides issus du K-means de la même couleur que ceux de l'image. Comment faire ?

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
plt.plot(WL,  KMEANS.cluster_centers_[0].T)
plt.ylabel('Raman intensity (UA)')
plt.title('Spectre cluster 1 ')
plt.plot(WL,  KMEANS.cluster_centers_[1].T)
plt.ylabel('Raman intensity (UA)')
plt.title('Spectre cluster 2 ')
plt.plot(WL,  KMEANS.cluster_centers_[2].T)
plt.ylabel('Raman intensity (UA)')
plt.title('Spectre cluster 3 ')
plt.show()

Merci !!!

Nicolas