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

Je ne suis pas sûr que cette fonctionnalité soit nouvelle, mais vous pouvez également utiliser la méthode get_legend_handles_labels() plutôt que de garder une 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()

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

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

ou

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

cela ne ferait qu'ajouter une étiquette à la légende de la hache.

je pense que c'est un moyen beaucoup plus facile. Il n'est pas nécessaire de suivre les lignes automatiquement lorsque vous avez seulement quelques lignes dans les second axes, car la fixation à la main comme ci-dessus serait assez facile. De toute façon, ça dépend de ce dont tu as besoin.

le le code entier est comme ci-dessous:

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()

la parcelle est comme suit:

mise à jour: ajouter une meilleure version:

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

cela ne fera rien alors que plot(0, 0) peut changer la plage des axes.

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

fig.legend(loc=1)

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=1)

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

plt.show()

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