Sumbu sekunder dengan twinx (): bagaimana cara menambahkan legenda?

Saya punya plot dengan dua sumbu y, menggunakan twinx(). Saya juga memberi label pada garis, dan ingin menunjukkannya legend(), tetapi saya hanya berhasil mendapatkan label satu sumbu dalam legenda:

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

Jadi saya hanya mendapatkan label sumbu pertama dalam legenda, dan bukan label 'temp' dari sumbu kedua. Bagaimana saya bisa menambahkan label ketiga ini ke legenda?

Anda dapat dengan mudah menambahkan legenda kedua dengan menambahkan baris:

ax2.legend(loc=0)

Anda akan mendapatkan ini:

Tetapi jika Anda ingin semua label pada satu legenda maka Anda harus melakukan sesuatu seperti ini:

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

Yang akan memberi Anda ini:

Saya tidak yakin apakah fungsi ini baru, tetapi Anda juga dapat menggunakan metode get_legend_handles_labels () daripada melacak sendiri baris dan label:

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

Dari matplotlib versi 2.1 dan seterusnya, Anda dapat menggunakan legenda gambar . Alih-alih ax.legend(), yang menghasilkan legenda dengan gagang dari kapak ax, orang dapat membuat tokoh legenda

fig.legend (loc = "kanan atas")

yang akan mengumpulkan semua pegangan dari semua subplot pada gambar. Karena itu adalah legenda tokoh, itu akan ditempatkan di sudut gambar, dan locargumen itu relatif terhadap sosok itu.

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

Untuk menempatkan legenda kembali ke kapak, orang akan memasok a bbox_to_anchordan a bbox_transform. Yang terakhir akan menjadi transformasi sumbu dari sumbu yang harus ditinggali legenda. Yang pertama mungkin adalah koordinat tepi yang ditentukan dengan locdiberikan dalam koordinat sumbu.

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

Anda dapat dengan mudah mendapatkan apa yang Anda inginkan dengan menambahkan baris dalam kapak:

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

atau

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

Ini tidak akan merencanakan apa pun selain menambahkan label ke legenda kapak.

Saya pikir ini adalah cara yang jauh lebih mudah. Tidak perlu melacak garis secara otomatis ketika Anda hanya memiliki beberapa garis di sumbu kedua, karena memperbaiki dengan tangan seperti di atas akan sangat mudah. Bagaimanapun, itu tergantung pada apa yang Anda butuhkan.

Seluruh kode adalah sebagai berikut:

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

Plotnya seperti di bawah ini:

Perbarui: tambahkan versi yang lebih baik:

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

Ini tidak akan melakukan apa-apa sementara plot(0, 0)dapat mengubah kisaran sumbu.

Satu contoh tambahan untuk pencar

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)

Retas cepat yang mungkin sesuai dengan kebutuhan Anda ..

Lepaskan bingkai kotak dan posisikan secara manual kedua legenda di samping satu sama lain. Sesuatu seperti ini..

ax1.legend(loc = (.75,.1), frameon = False)
ax2.legend( loc = (.75, .05), frameon = False)

Di mana loc locup adalah persentase dari kiri ke kanan dan dari bawah ke atas yang mewakili lokasi dalam bagan.

Saya menemukan contoh resmi matplotlib berikut yang menggunakan host_subplot untuk menampilkan banyak sumbu y dan semua label berbeda dalam satu legenda. Tidak ada solusi yang diperlukan. Solusi terbaik yang saya temukan sejauh ini. http://matplotlib.org/examples/axes_grid/demo_parasite_axes2.html

from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import matplotlib.pyplot as plt

host = host_subplot(111, axes_class=AA.Axes)
plt.subplots_adjust(right=0.75)

par1 = host.twinx()
par2 = host.twinx()

offset = 60
new_fixed_axis = par2.get_grid_helper().new_fixed_axis
par2.axis["right"] = new_fixed_axis(loc="right",
                                    axes=par2,
                                    offset=(offset, 0))

par2.axis["right"].toggle(all=True)

host.set_xlim(0, 2)
host.set_ylim(0, 2)

host.set_xlabel("Distance")
host.set_ylabel("Density")
par1.set_ylabel("Temperature")
par2.set_ylabel("Velocity")

p1, = host.plot([0, 1, 2], [0, 1, 2], label="Density")
p2, = par1.plot([0, 1, 2], [0, 3, 2], label="Temperature")
p3, = par2.plot([0, 1, 2], [50, 30, 15], label="Velocity")

par1.set_ylim(0, 4)
par2.set_ylim(1, 65)

host.legend()

plt.draw()
plt.show()