come si crea una singola legenda per molte sottotrame con matplotlib?

Question

Sto tracciando lo stesso tipo di informazioni, ma per paesi diversi, con sottotrame multiple con matplotlib. Cioè, ho 9 grafici su una griglia 3x3, tutti con lo stesso per le linee (ovviamente, valori diversi per linea).

Tuttavia, non ho capito come mettere una sola legenda (dato che tutte e 9 le sottotrame hanno le stesse linee) sulla figura solo una volta.

Come posso fare?

Answer 1

figlegend può essere quello che stai cercando: http://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.figlegend

Esempio qui: http://matplotlib.org/examples/pylab_examples/figlegend_demo.html

Un altro esempio:

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

o:

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

Answer 2

Devi solo chiedi la leggenda una volta, al di fuori del tuo ciclo.

Ad esempio, in questo caso ho 4 sottotrame, con le stesse linee e una singola legenda.

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

Per il posizionamento automatico di un singolo legenda in una figure con molti assi, come quelli ottenuti con subplots(), la seguente soluzione funziona molto bene:

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

Con bbox_to_anchor e bbox_transform=plt.gcf().transFigure si definisce un nuovo riquadro di delimitazione delle dimensioni del tuo figure come riferimento per loc. L'utilizzo di (0,-0.1,1,1) sposta questo riquadro verso il basso leggermente per evitare che la legenda venga posizionata su altri artisti.

OBS: utilizzare questa soluzione dopo aver utilizzato fig.set_size_inches() e prima di utilizzare fig.tight_layout()

Answer 4

C'è anche una bella funzione get_legend_handles_labels() è possibile chiamare l'ultimo asse (se di eseguire iterazioni su di loro) che raccolga tutto il necessario da label= argomenti:

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

Answer 5

Questa risposta è un complemento @ Evert è sulla posizione di leggenda.

Il mio primo tentativo su @ soluzione di Evert è fallito a causa di sovrapposizioni della leggenda e del titolo della sottotrama.

In realtà, le sovrapposizioni sono causate da fig.tight_layout(), che modifica il layout delle sottotrame senza considerare la legenda della figura. Tuttavia, è necessario fig.tight_layout().

Per evitare sovrapposizioni, è possibile comunicare allo fig.tight_layout() di lasciare spazi per la legenda della figura tramite fig.tight_layout(rect=(0,0,1,0.9)).

Description of tight_layout() parameters.