Gráfico de barras pandas con colores específicos y ubicación de la leyenda?

Question

Tengo un pandas DataFrame y quiero trazar un gráfico de barras que incluya una leyenda.

import pylab as pl 
from pandas import * 

x = DataFrame({"Alpha": Series({1: 1, 2: 3, 3:2.5}), "Beta": Series({1: 2, 2: 2, 3:3.5})}) 

Si llamo directamente trama, entonces se pone la leyenda por encima de la trama:

x.plot(kind="bar") 

Si giro de la leyenda en la trama y tratar de añadirlo más tarde, entonces no lo hace conservar los colores asociados a las dos columnas de la trama de datos (véase más adelante):

x.plot(kind="bar", legend=False) 
l = pl.legend(('Alpha','Beta'), loc='best') 

¿Cuál es la forma correcta de incluir una leyenda en una parcela matplotlib de una trama de datos pandas?

Answer 1

Si desea agregar la leyenda manualmente, tiene que pedir a la subtrama de los elementos del diagrama de barras:

In [17]: ax = x.plot(kind='bar', legend=False) 

In [18]: patches, labels = ax.get_legend_handles_labels() 

In [19]: ax.legend(patches, labels, loc='best') 
Out[19]: <matplotlib.legend.Legend at 0x10b292ad0> 

Además, plt.legend(loc='best') o ax.legend(loc='best') debe "sólo trabajo", porque hay ya se "vincula" a los parches del diagrama de barras configurados cuando se realiza el trazado, por lo que no es necesario pasar una lista de etiquetas de ejes.

No estoy seguro de si la versión de pandas que está usando devuelve un identificador a la subtrama (ax = ...) pero estoy bastante seguro de que 0.7.3 sí lo hace. Siempre puede obtener una referencia con plt.gca().

Answer 2

La forma más sucinta que ir es:

x.plot(kind="bar").legend(bbox_to_anchor=(1.2, 0.5)) 

o en general

x.plot(kind="bar").legend(*args, **kwargs)