Problème affichage couleur heatmap

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:

---

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:

---

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