Legenda Matplotlib, aggiungi elementi su colonne invece che verso il basso

Question

Per una trama semplice di seguito, c'è un modo per rendere matplotlib popolare la legenda in modo che riempia le righe da sinistra a destra, anziché prima colonna poi seconda colonna?

>>> from pylab import * 
>>> x = arange(-2*pi, 2*pi, 0.1) 
>>> plot(x, sin(x), label='Sine') 
>>> plot(x, cos(x), label='Cosine') 
>>> plot(x, arctan(x), label='Inverse tan') 
>>> legend(loc=9,ncol=2) 
>>> grid('on') 

Answer 1

mi viene in mente un modo possibile. Puoi order your legend items come preferisci. Tutto quello che devi fare è cambiare l'ordine in modo che ti dia il risultato che desideri.

import matplotlib.pyplot as plt 
import numpy as np 
import itertools 

def flip(items, ncol): 
    return itertools.chain(*[items[i::ncol] for i in range(ncol)]) 

x = np.arange(-2*np.pi, 2*np.pi, 0.1) 
ax = plt.subplot(111) 
ax.plot(x, np.sin(x), label='Sine') 
ax.plot(x, np.cos(x), label='Cosine') 
ax.plot(x, np.arctan(x), label='Inverse tan') 

handles, labels = ax.get_legend_handles_labels() 
plt.legend(flip(handles, 2), flip(labels, 2), loc=9, ncol=2) 

plt.grid('on') 
plt.show()