Comment créer des nuages ​​de points avec des couleurs distinctes pour des niveaux catégoriels en Python à l'aide de Matplotlib ?

Dessiner des nuages ​​de points avec différentes couleurs pour des niveaux catégoriels en Python avec Matplotlib

Dans Matplotlib, une bibliothèque Python permettant de créer des images statiques, animées et visualisations interactives en Python, vous pouvez tracer différents nuages ​​de points avec des couleurs différentes pour chaque niveau d'une variable catégorielle en tirant parti de l'argument c de la fonction plt.scatter.

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

df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6], 'color': ['red', 'blue', 'green']})

colors = {'red': 'tab:red', 'blue': 'tab:blue', 'green': 'tab:green'}

plt.scatter(df['x'], df['y'], c=df['color'].map(colors))
plt.show()</code>

En passant l'argument c, un dictionnaire le mappage des noms de couleurs aux valeurs RVB peut être utilisé pour spécifier la couleur de chaque point. La méthode cartographique de Pandas applique ensuite le mappage des couleurs à la colonne df['color'], attribuant efficacement à chaque point une couleur unique.

<code class="python">colors = {'D': 'tab:blue', 'E': 'tab:orange', 'F': 'tab:green', 'G': 'tab:red', 'H': 'tab:purple', 'I': 'tab:brown', 'J': 'tab:pink'}

ax.scatter(df['carat'], df['price'], c=df['color'].map(colors))</code>

Cette approche permet une palette de couleurs plus personnalisée et un meilleur contrôle sur les couleurs utilisées dans l'intrigue. En utilisant un dictionnaire de couleurs, les utilisateurs peuvent facilement modifier la palette de couleurs selon leurs besoins.

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!