Axe secondaire avec twinx() : comment ajouter à la légende ?

Question

J'ai un tracé avec deux axes y, utilisant twinx() . Je donne également des étiquettes aux lignes, et je veux les afficher avec legend() mais je ne parviens à faire apparaître les étiquettes d'un seul axe dans la légende :

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
ax2.plot(time, temp, '-r', label = 'temp')
ax.legend(loc=0)
ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

Je n'obtiens donc que les étiquettes du premier axe dans la légende, et pas l'étiquette "temp" du deuxième axe. Comment puis-je ajouter cette troisième étiquette à la légende ?

Answer 1

Vous pouvez facilement ajouter une deuxième légende en ajoutant la ligne :

ax2.legend(loc=0)

Vous aurez ça :

Mais si vous voulez toutes les étiquettes sur une seule légende, vous devez faire quelque chose comme ceci :

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

time = np.arange(10)
temp = np.random.random(10)*30
Swdown = np.random.random(10)*100-10
Rn = np.random.random(10)*100-10

fig = plt.figure()
ax = fig.add_subplot(111)

lns1 = ax.plot(time, Swdown, '-', label = 'Swdown')
lns2 = ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
lns3 = ax2.plot(time, temp, '-r', label = 'temp')

# added these three lines
lns = lns1+lns2+lns3
labs = [l.get_label() for l in lns]
ax.legend(lns, labs, loc=0)

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

Ce qui vous donnera ceci :

Answer 2

Je ne sais pas si cette fonctionnalité est nouvelle, mais vous pouvez également utiliser la méthode get_legend_handles_labels() plutôt que de garder la trace des lignes et des étiquettes vous-même :

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

pi = np.pi

# fake data
time = np.linspace (0, 25, 50)
temp = 50 / np.sqrt (2 * pi * 3**2) \
        * np.exp (-((time - 13)**2 / (3**2))**2) + 15
Swdown = 400 / np.sqrt (2 * pi * 3**2) * np.exp (-((time - 13)**2 / (3**2))**2)
Rn = Swdown - 10

fig = plt.figure()
ax = fig.add_subplot(111)

ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')
ax2 = ax.twinx()
ax2.plot(time, temp, '-r', label = 'temp')

# ask matplotlib for the plotted objects and their labels
lines, labels = ax.get_legend_handles_labels()
lines2, labels2 = ax2.get_legend_handles_labels()
ax2.legend(lines + lines2, labels + labels2, loc=0)

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

Answer 3

À partir de la version 2.1 de matplotlib, vous pouvez utiliser une balise légende des figures . Au lieu de ax.legend() qui produit une légende avec les poignées des axes. ax on peut créer une légende des chiffres

fig.legend(loc="upper right")

qui rassemblera toutes les poignées de tous les sous-plots de la figure. Puisqu'il s'agit d'une légende de figure, elle sera placée dans le coin de la figure, et l'attribut loc L'argument est relatif à la figure.

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0,10)
y = np.linspace(0,10)
z = np.sin(x/3)**2*98

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(x,y, '-', label = 'Quantity 1')

ax2 = ax.twinx()
ax2.plot(x,z, '-r', label = 'Quantity 2')
fig.legend(loc="upper right")

ax.set_xlabel("x [units]")
ax.set_ylabel(r"Quantity 1")
ax2.set_ylabel(r"Quantity 2")

plt.show()

Afin de replacer la légende dans les axes, il faut fournir un fichier bbox_to_anchor y un bbox_transform . Ce dernier serait la transformée des axes dans lesquels la légende devrait résider. Le premier peut être les coordonnées du bord défini par loc donné en coordonnées d'axes.

fig.legend(loc="upper right", bbox_to_anchor=(1,1), bbox_transform=ax.transAxes)

Answer 4

Vous pouvez facilement obtenir ce que vous voulez en ajoutant la ligne en ax :

ax.plot([], [], '-r', label = 'temp')

o

ax.plot(np.nan, '-r', label = 'temp')

Cela ne ferait rien d'autre que d'ajouter une étiquette à la légende de l'axe.

Je pense que c'est un moyen beaucoup plus facile. Il n'est pas nécessaire de suivre les lignes automatiquement lorsque vous n'avez que quelques lignes dans le second axe, car il serait assez facile de les fixer à la main comme ci-dessus. Quoi qu'il en soit, cela dépend de ce dont vous avez besoin.

Le code complet est le suivant :

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('mathtext', default='regular')

time = np.arange(22.)
temp = 20*np.random.rand(22)
Swdown = 10*np.random.randn(22)+40
Rn = 40*np.random.rand(22)

fig = plt.figure()
ax = fig.add_subplot(111)
ax2 = ax.twinx()

#---------- look at below -----------

ax.plot(time, Swdown, '-', label = 'Swdown')
ax.plot(time, Rn, '-', label = 'Rn')

ax2.plot(time, temp, '-r')  # The true line in ax2
ax.plot(np.nan, '-r', label = 'temp')  # Make an agent in ax

ax.legend(loc=0)

#---------------done-----------------

ax.grid()
ax.set_xlabel("Time (h)")
ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")
ax2.set_ylabel(r"Temperature ($^\circ$C)")
ax2.set_ylim(0, 35)
ax.set_ylim(-20,100)
plt.show()

L'intrigue est la suivante :

---

Mise à jour : ajouter une meilleure version :

ax.plot(np.nan, '-r', label = 'temp')

Cela ne fera rien pendant que plot(0, 0) peut modifier la portée de l'axe.

---

Un exemple supplémentaire de dispersion

ax.scatter([], [], s=100, label = 'temp')  # Make an agent in ax
ax2.scatter(time, temp, s=10)  # The true scatter in ax2

ax.legend(loc=1, framealpha=1)

Answer 5

Préparation

import numpy as np
from matplotlib import pyplot as plt

fig, ax1 = plt.subplots( figsize=(15,6) )

Y1, Y2 = np.random.random((2,100))

ax2 = ax1.twinx()

Contenu

Je suis surpris que cela ne soit pas apparu jusqu'à présent, mais le plus simple est de les rassembler manuellement dans l'un des objets d'axes (qui se trouvent les uns sur les autres).

l1 = ax1.plot( range(len(Y1)), Y1, label='Label 1' )
l2 = ax2.plot( range(len(Y2)), Y2, label='Label 2', color='orange' )

ax1.legend( handles=l1+l2 )

ou les faire collecter automatiquement dans la figure environnante par fig.legend() et manipuler les bbox_to_anchor paramètre :

ax1.plot( range(len(Y1)), Y1, label='Label 1' )
ax2.plot( range(len(Y2)), Y2, label='Label 2', color='orange' )

fig.legend( bbox_to_anchor=(.97, .97) )

Finalisation

fig.tight_layout()
fig.savefig('stackoverflow.png', bbox_inches='tight')