Comment donner une pandas/matplotlib graphique à barres de couleurs personnalisées

J'ai juste commencé à l'aide de pandas/matplotlib comme un remplacement pour Excel pour générer des graphiques à barres empilées. Je suis en cours d'exécution dans un problème

(1) il y a seulement 5 couleurs dans la palette par défaut, donc si j'ai plus de 5 catégories puis les couleurs de répétition. Comment puis-je spécifier plus de couleurs? Idéalement, un dégradé avec une couleur de début et de fin de couleur, et un moyen pour générer dynamiquement n couleurs entre les deux?

(2) les couleurs ne sont pas très agréable visuellement. Comment puis-je spécifier un ensemble personnalisé de n couleurs? Ou, un gradient serait aussi travailler.

Un exemple qui illustre à la fois des points ci-dessus est ci-dessous:

  4 from matplotlib import pyplot
  5 from pandas import *
  6 import random
  7 
  8 x = [{i:random.randint(1,5)} for i in range(10)]
  9 df = DataFrame(x)
 10 
 11 df.plot(kind='bar', stacked=True)

Et la sortie est ceci:

Comment donner une pandas/matplotlib graphique à barres de couleurs personnalisées

Il y a une manière assez facile d'obtenir une partielle palette de couleurs. Voir cette solution ci-dessous

InformationsquelleAutor vasek1 | 2012-08-13

88

Vous pouvez spécifier le color option d'une liste directement à la plot fonction.
```
from matplotlib import pyplot as plt
from itertools import cycle, islice
import pandas, numpy as np  # I find np.random.randint to be better

# Make the data
x = [{i:np.random.randint(1,5)} for i in range(10)]
df = pandas.DataFrame(x)

# Make a list by cycling through the colors you care about
# to match the length of your data.
my_colors = list(islice(cycle(['b', 'r', 'g', 'y', 'k']), None, len(df)))

# Specify this list of colors as the `color` option to `plot`.
df.plot(kind='bar', stacked=True, color=my_colors)
```
Pour définir votre propre liste personnalisée, vous pouvez faire un peu de la suite, ou il suffit de regarder la Matplotlib techniques pour la définition d'un élément de couleur par ses valeurs RVB, etc. Vous pouvez obtenir d'aussi compliqué que vous voulez avec ce.
```
my_colors = ['g', 'b']*5 # <-- this concatenates the list to itself 5 times.
my_colors = [(0.5,0.4,0.5), (0.75, 0.75, 0.25)]*5 # <-- make two custom RGBs and repeat/alternate them over all the bar elements.
my_colors = [(x/10.0, x/20.0, 0.75) for x in range(len(df))] # <-- Quick gradient example along the Red/Green dimensions.
```
Le dernier exemple, les rendements de la suivre simple dégradé de couleurs pour moi:

Je n'ai pas jouer avec elle assez longtemps pour comprendre comment la force de la légende et de ramasser les couleurs définies, mais je suis sûr que vous pouvez le faire.

En général, cependant, un gros morceau de conseil est simplement d'utiliser les fonctions de Matplotlib directement. L'appel de Pandas est OK, mais je trouve que vous obtenez de meilleures options de performance et de les appeler directement à partir de Matplotlib.
- Bug mineur: my_colors = [cycle(['b', 'r', 'g', 'y', 'k']).next() for i in range(len(df))] donne " b " à chaque fois, dans python 2.7. Vous devez utiliser la liste(islice(cycle(['b', 'r', 'g', 'y', 'k']), None, len(df))) à la place.
- Merci, je n'aurais probablement pas pris que l'. Une autre option est de créer le premier cycle, puis il suffit d'appeler son next fonction à l'intérieur de la compréhension.
- Yup. c' = cycle(['b', 'r', 'g', 'y', 'k']); my_colors=[suivant(il) for i in xrange(len(df))] serait de couper ainsi...
- Avec les pandas et matplotlib installé aujourd'hui, le code ci-dessus n'engendre rien pour moi, bien qu'il s'exécute.
- Êtes-vous en cours d'exécution dans l'ordinaire de l'interprète, IPython, ou à partir de l'interpréteur de commandes (ou autre chose)? Selon le type d'environnement que vous exécutez ce code, vous pourriez avoir besoin pour activer le mode interactif pour matplotlib, ou un ensemble pylab.ion() interactif pylab.
- this concatenates the list to itself 5 times. Déjà cycles par le biais de la liste que vous donnez; vous n'avez pas besoin de le faire manuellement
- Ouais ce sont juste des exemples rapides. La partie sur la concaténation de la liste elle-même a zéro portant sur cette question ou d'une réponse.
- est-il un moyen de mettre la "couleur=my_colors" ce dans rcParams ?
- Si vous utilisez seaborn, vous pouvez aussi avoir générer une liste de couleurs avec les sns.color_palette("Set2", 10) pour les pandas/matplotlib de cycle jusqu'à, par exemple, df.plot.area(stacked=True, color=sns.color_palette('Set3', 12))
InformationsquelleAutor ely
40

J'ai trouvé le moyen le plus facile est d'utiliser le colormap paramètre dans .plot() avec l'un des programmes de dégradés de couleurs:
```
df.plot(kind='bar', stacked=True, colormap='Paired')
```
Vous pouvez trouver une grande liste de présélection des colormaps ici.
- dans mon cas ce seulement donne une couleur à travers tous les bars
InformationsquelleAutor Alexandre
10

Pour une réponse plus détaillée sur la création de votre propre carte des couleurs, je vous suggère fortement de visiter cette page

Si la réponse est trop de travail, vous pouvez rapidement faire votre propre liste de couleurs et de les transmettre à l' color paramètre. Tous les réglages de palettes de couleur sont dans la cm module matplotlib. Nous allons obtenir une liste de 30 RGB (plus alpha) les valeurs de couleur de l'inversion de l'enfer palette de couleurs. Pour ce faire, la première à obtenir la palette de couleurs et de transmettre une séquence de valeurs entre 0 et 1. Ici, nous utilisons np.linspace créer 30 équidistants des valeurs comprises entre 4 et 8 la qui représentent la partie de la palette de couleurs.
```
from matplotlib import cm
color = cm.inferno_r(np.linspace(.4,.8, 30))
color

array([[ 0.865006,  0.316822,  0.226055,  1.      ],
       [ 0.851384,  0.30226 ,  0.239636,  1.      ],
       [ 0.832299,  0.283913,  0.257383,  1.      ],
       [ 0.817341,  0.270954,  0.27039 ,  1.      ],
       [ 0.796607,  0.254728,  0.287264,  1.      ],
       [ 0.775059,  0.239667,  0.303526,  1.      ],
       [ 0.758422,  0.229097,  0.315266,  1.      ],
       [ 0.735683,  0.215906,  0.330245,  1.      ],
       .....
```
Alors nous pouvons l'utiliser à la parcelle - en utilisant les données de l'original post:
```
import random
x = [{i:random.randint(1,5)} for i in range(30)]
df = pd.DataFrame(x)
df.plot(kind='bar', stacked=True, color=color, legend=False, figsize=(12,4))
```
- Cette carte de la couleur a l'air très bien. Merci pour la réponse!
- Voici la documentation pour les autres cartes de couleurs en dehors de inferno_r: matplotlib.org/examples/color/colormaps_reference.html
- J'ai suivi cet extrait, mais ma gamme de couleurs a toujours les mêmes valeurs.
InformationsquelleAutor Ted Petrou

Vous devez vous connecter pour publier un commentaire.