Afficher la barre de couleurs dans Matplotlib

Dans ce didacticiel, nous verrons comment afficher une barre de couleurs dans une figure matplotlib en Python.

Pour créer une barre de couleurs, nous devons utiliser la fonction matplotlib.pyplot.colorbar().

Le code suivant montre un exemple simple de ceci.

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

s = plt.scatter(s_x, s_y, c=s_x, cmap="viridis")

c = plt.colorbar()

Dans l’exemple ci-dessus, nous avons créé une simple barre de couleurs placée à l’extérieur du tracé. Nous avons spécifié la palette de couleurs à l’aide du paramètre cmap.

Nous pouvons également spécifier les axes dans lesquels nous souhaitons afficher la barre de couleur. Si nous le souhaitons, nous pouvons l’ajouter sur les axes du tracé.

Par example,

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

fig, ax = plt.subplots()
cax = fig.add_axes([0.27, 0.8, 0.5, 0.05])

im = ax.scatter(s_x, s_y, c=s_x, cmap="viridis")
fig.colorbar(im, cax=cax, orientation="horizontal")

Dans l’exemple ci-dessus, nous avons utilisé la fonction subplots() pour obtenir les objets figure et axes et l’utiliser pour créer les axes de la barre de couleurs. Nous l’avons spécifié en utilisant le paramètre cax dans la fonction colorbar().

A noter également l’utilisation du paramètre orientation qui a modifié l’orientation de la barre de couleur finale. En dehors de cela, nous pouvons utiliser différents paramètres pour contrôler la forme et la taille de la barre de couleurs. Par exemple, shrink peut réduire la taille de la barre de couleur d’une petite marge, aspect, qui est le rapport des côtés de la barre, et il y en a bien d’autres.