Comment créer une barre de couleurs discrète personnalisée pour un nuage de points Matplotlib ?

Barre de couleurs discrète personnalisée pour le nuage de points Matplotlib

Dans Matplotlib, la création d'une barre de couleurs discrète pour un nuage de points vous permet de représenter des points de données avec des couleurs uniques basées sur sur leurs valeurs de balise entières. Pour y parvenir, une carte de couleurs discrète personnalisée et une BoundaryNorm peuvent être utilisées.

Construisons un nuage de points avec des données x et y aléatoires et attribuons des balises entières allant de 0 à 20 :

<code class="python">import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(20)
y = np.random.rand(20)
tag = np.random.randint(0, 20, 20)</code>

Initialement, en utilisant les paramètres par défaut, la barre de couleurs affiche une gamme continue de couleurs :

<code class="python">plt.scatter(x, y, c=tag)
plt.colorbar()</code>

Pour créer une barre de couleurs discrète, nous utiliserons une BoundaryNorm pour définir les limites de la palette de couleurs. Nous voulons également nous assurer que la valeur de balise 0 est représentée par une couleur grise :

<code class="python">bounds = np.linspace(0, 20, 21)
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)

cmaplist = [cmap(i) for i in range(cmap.N)]
cmaplist[0] = (.5, .5, .5, 1.0)

cmap = mpl.colors.LinearSegmentedColormap.from_list(
    'Custom cmap', cmaplist, cmap.N)</code>

Le nuage de points mis à jour avec la barre de couleurs discrète ressemble à ceci :

<code class="python">scat = ax.scatter(x, y, c=tag, s=np.random.randint(100, 500, 20),
                  cmap=cmap, norm=norm)

# Add the discrete colorbar
ax2 = fig.add_axes([0.95, 0.1, 0.03, 0.8])
cb = plt.colorbar.ColorbarBase(ax2, cmap=cmap, norm=norm,
    spacing='proportional', ticks=bounds, boundaries=bounds, format='%1i')</code>

Cette approche fournit une barre de couleurs discrète bien définie pour votre nuage de points, où chaque valeur de balise est représentée par une couleur unique, y compris une couleur grise pour la valeur de balise 0.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!