IP Girl
  1. IP Girl
  3. matplotlib: 2 légendes différentes sur le même graphique
Intereting Posts
Comment peut-on supprimer une contrainte non nulle? Comment utiliser la bibliothèque v7 / v14 Preference Support? Qu’est-ce que orderInCategory dans l’élément de menu ActionBar et pourquoi est-il utilisé pour ..? Comment append des parameters à HttpURLConnection à l’aide de POST en utilisant NameValuePair Paramètre de valeur de table de procédure stockée Entity Framework Comment puis-je rendre le texte vertical (pivoté à 90 degrés) en mode de réaction natif? Liste toutes les sessions ASP.NET actives managedQuery () vs context.getContentResolver.query () vs android.provider.something.query () Hibernate, @SequenceGenerator et allocationSize Supprimer le style de numéro de téléphone de Microsoft Edge Comment puis-je utiliser ClearCase pour «append au contrôle de source…» de manière récursive? Method vs Computed in Vue Comment puis-je exécuter PowerShell avec le runtime .NET 4? Quand devrions-nous utiliser la méthode intern de Ssortingng sur les littéraux Ssortingng DbContext est-il identique à DataContext?

matplotlib: 2 légendes différentes sur le même graphique

J’ai un tracé où différentes couleurs sont utilisées pour différents parameters et où différents styles de lignes sont utilisés pour différents algorithmes. Le but est de comparer les résultats des différents algorithmes réalisés avec des parameters similaires. Cela signifie qu’au total, j’utilise 4 couleurs différentes et 3 styles de lignes différents, pour un total de 12 tracés sur le même graphique.

Je construis en fait la légende en fonction des couleurs, en associant chaque couleur au paramètre correspondant. Maintenant, je voudrais afficher une deuxième légende sur le même graphique, avec la signification de chaque style de ligne. Il est possible d’y parvenir? Comment?

Voici à quoi ressemble mon code: 

colors = ['b', 'r', 'g', 'c'] cc = cycle(c) for p in parameters: d1 = algo1(p) d2 = algo2(p) d3 = algo3(p) pyplot.hold(True) c = next(cc) pyplot.plot(d1, '-', color=c, label="d1") pyplot.plot(d1, '--', color=c) pyplot.plot(d2, '.-', color=c) pyplot.legend()

Il y a une section dans la documentation de matplotlib sur ce sujet exact: http://matplotlib.org/users/legend_guide.html#multiple-legends-on-the-same-axes

Voici le code pour votre exemple spécifique: 

 import itertools from matplotlib import pyplot colors = ['b', 'r', 'g', 'c'] cc = itertools.cycle(colors) plot_lines = [] for p in parameters: d1 = algo1(p) d2 = algo2(p) d3 = algo3(p) pyplot.hold(True) c = next(cc) l1, = pyplot.plot(d1, '-', color=c) l2, = pyplot.plot(d2, '--', color=c) l3, = pyplot.plot(d3, '.-', color=c) plot_lines.append([l1, l2, l3]) legend1 = pyplot.legend(plot_lines[0], ["algo1", "algo2", "algo3"], loc=1) pyplot.legend([l[0] for l in plot_lines], parameters, loc=4) pyplot.gca().add_artist(legend1)

Voici un exemple de sa sortie: Terrain avec 2 légendes, par paramètre et par algo

Voici aussi une manière plus pratique de le faire (c.-à-d. Interagir explicitement avec n’importe quel axe de figure): 

 import itertools from matplotlib import pyplot fig, axes = plt.subplot(1,1) colors = ['b', 'r', 'g', 'c'] cc = itertools.cycle(colors) plot_lines = [] for p in parameters: d1 = algo1(p) d2 = algo2(p) d3 = algo3(p) c = next(cc) axes.plot(d1, '-', color=c) axes.plot(d2, '--', color=c) axes.plot(d3, '.-', color=c) # In total 3x3 lines have been plotted lines = axes.get_lines() legend1 = pyplot.legend([lines[i] for i in [0,1,2]], ["algo1", "algo2", "algo3"], loc=1) legend2 = pyplot.legend([lines[i] for i in [0,3,6]], parameters, loc=4) axes.add_artist(legend1) axes.add_artist(legend2)

J’aime cette façon de l’écrire car elle permet potentiellement de jouer avec différents axes de manière moins obscure. Vous pouvez d’abord créer votre ensemble de légendes, puis les append aux axes de votre choix avec la méthode “add_artist”. De plus, je commence avec matplotlib, et pour moi au moins, il est plus facile de comprendre les scripts lorsque des objects sont expliqués.

NB: Attention, vos légendes peuvent être coupées lors de l’affichage / sauvegarde. Pour résoudre ce problème, utilisez la méthode axes.set_position ([left, bottom, width, length]) pour réduire la sous-insortinggue relativement à la taille de la figure et faire apparaître les légendes.