Heatmap

Représentation d'une matrice sous forme de heatmap (un carré par valeur, et une couleur qui dépend de cette valeur) :

• mat = [[1, 2, 3, 1], [4, 5, 6, 4], [7, 7, 10, 10]]; pyplot.matshow(mat) : les lignes de la matrice sont représentées en ligne et les colonnes en colonnes, et la première ligne est en haut (correspond à la représentation matricielle habituelle).
• origin = 'lower' : la première ligne est mise tout en bas et la dernière tout en haut.
• cmap = pyplot.cm.gist_rainbow : indique une carte des couleurs alternative.
• pyplot.xticks(range(4), ['A', 'B', 'C', 'D']) : permet d'indiquer des labels pour chaque colonne (pareil avec yticks pour les lignes).
• si on veut avoir 2 heatmaps sur une seule figure, passer par la forme objet (cf exemple ci-dessous).

mat = [[1, 2, 3, 1], [4, 5, 6, 4], [7, 7, 10, 10]] pyplot.matshow(mat) pyplot.title('exemple de matrice avec couleurs par defaut')

Exemple avec 2 heatmaps sur la même figure : figure = pyplot.figure() axes = figure.add_subplot(2, 1, 1) mat = [[1, 2, 3, 1], [4, 5, 6, 4], [8, 7, 10, 9]] axes.matshow(mat, cmap = pyplot.cm.gist_rainbow) axes.set_xticks(range(4)) axes.set_xticklabels(['A', 'B', 'C', 'D']) axes.set_yticks(range(3)) axes.set_yticklabels(['a', 'b', 'c']) axes.set_title('origine en haut a gauche') axes = figure.add_subplot(2, 1, 2) axes.matshow(mat, origin = 'lower', cmap = pyplot.cm.gist_rainbow) axes.set_xticks(range(4)) axes.set_xticklabels(['A', 'B', 'C', 'D']) axes.set_yticks(range(3)) axes.set_yticklabels(['a', 'b', 'c']) axes.set_title('origine en bas a gauche')