¿cómo hago una sola leyenda para muchas subtramas con matplotlib?

Question

Estoy trazando el mismo tipo de información, pero para diferentes países, con múltiples subtramas con matplotlib. Es decir, tengo 9 tramas en una cuadrícula de 3x3, todas con lo mismo para las líneas (por supuesto, diferentes valores por línea).

Sin embargo, no he descubierto cómo poner una sola leyenda (ya que las 9 subtramas tienen las mismas líneas) en la figura solo una vez.

¿Cómo puedo hacer eso?

Answer 1

figlegend puede ser lo que está buscando: http://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.figlegend

Ejemplo aquí: http://matplotlib.org/examples/pylab_examples/figlegend_demo.html

Otro ejemplo:

plt.figlegend(lines, labels, loc = 'lower center', ncol=5, labelspacing=0.) 

o:

fig.legend(lines, labels, loc = (0.5, 0), ncol=5) 

Answer 2

Sólo hay que pregunta por la leyenda una vez, fuera de tu loop.

Por ejemplo, en este caso tengo 4 subtramas, con las mismas líneas, y una sola leyenda.

from matplotlib.pyplot import * 

ficheiros = ['120318.nc', '120319.nc', '120320.nc', '120321.nc'] 

fig = figure() 
fig.suptitle('concentration profile analysis') 

for a in range(len(ficheiros)): 
    # dados is here defined 
    level = dados.variables['level'][:] 

    ax = fig.add_subplot(2,2,a+1) 
    xticks(range(8), ['0h','3h','6h','9h','12h','15h','18h','21h']) 
    ax.set_xlabel('time (hours)') 
    ax.set_ylabel('CONC ($\mu g. m^{-3}$)') 

    for index in range(len(level)): 
     conc = dados.variables['CONC'][4:12,index] * 1e9 
     ax.plot(conc,label=str(level[index])+'m') 

    dados.close() 

ax.legend(bbox_to_anchor=(1.05, 0), loc='lower left', borderaxespad=0.) 
     # it will place the legend on the outer right-hand side of the last axes 

show() 

Answer 3

Para la colocación automática de una sola leyenda en un figure con muchos ejes, como los obtenidos con subplots(), la siguiente solución funciona muy bien:

plt.legend(lines, labels, loc = 'lower center', bbox_to_anchor = (0,-0.1,1,1), 
      bbox_transform = plt.gcf().transFigure) 

Con bbox_to_anchor y bbox_transform=plt.gcf().transFigure está definiendo una nuevo cuadro delimitador del tamaño de su figure como referencia para loc. El uso de (0,-0.1,1,1) mueve esta caja de dibujo ligeramente hacia abajo para evitar que la leyenda se coloque sobre otros artistas.

OBS: utilizar esta solución después de que utiliza fig.set_size_inches() y antes del uso fig.tight_layout()

Answer 4

También hay una función agradable get_legend_handles_labels() puede llamar en el último eje (si iterar sobre ellos) que sería recoger todo lo que necesita de label= argumentos:

handles, labels = ax.get_legend_handles_labels() 
fig.legend(handles, labels, loc='upper center') 

Answer 5

Esta respuesta es un complemento a @ Evert está en la posición de la leyenda.

Mi primera solución try @ Evert falló debido a la superposición de la leyenda y el título de la subtrama.

De hecho, las superposiciones son causadas por fig.tight_layout(), que cambia el diseño de las subtramas sin tener en cuenta la leyenda de la figura. Sin embargo, es necesario fig.tight_layout().

Para evitar las superposiciones, podemos decir fig.tight_layout() para dejar espacios para la leyenda de la figura por fig.tight_layout(rect=(0,0,1,0.9)).

Description of tight_layout() parameters.