Adăugarea unei legende la PyPlot în Matplotlib în cel mai simplu mod posibil

TL;DR -> Cum se poate crea o legendă pentru un grafic linie în Matplotlib's PyPlot fără a crea orice variabile suplimentare?

Vă rugăm să ia în considerare grafice script-ul de mai jos:

if __name__ == '__main__':
    PyPlot.plot(total_lengths, sort_times_bubble, 'b-',
                total_lengths, sort_times_ins, 'r-',
                total_lengths, sort_times_merge_r, 'g+',
                total_lengths, sort_times_merge_i, 'p-', )
    PyPlot.title("Combined Statistics")
    PyPlot.xlabel("Length of list (number)")
    PyPlot.ylabel("Time taken (seconds)")
    PyPlot.show()

După cum puteți vedea, aceasta este o foarte de bază de utilizare a matplotlib'sPyPlot`. Acest ideal generează un grafic ca cel de mai jos:

Nimic special, știu. Cu toate acestea, nu este clar ce fel de date este de a fi complotat în cazul în care (I'm încercat să complot datele unor algoritmi de sortare, lungime împotriva dată luate, și am'd dori să asigurați-vă că oamenii știu care este linia care). Astfel, am nevoie de o legenda, cu toate acestea, de a lua o privire la următorul exemplu de mai jos(de la site-ul oficial):

ax = subplot(1,1,1)
p1, = ax.plot([1,2,3], label="line 1")
p2, = ax.plot([3,2,1], label="line 2")
p3, = ax.plot([2,3,1], label="line 3")

handles, labels = ax.get_legend_handles_labels()

# reverse the order
ax.legend(handles[::-1], labels[::-1])

# or sort them by labels
import operator
hl = sorted(zip(handles, labels),
            key=operator.itemgetter(1))
handles2, labels2 = zip(*hl)

ax.legend(handles2, labels2)

Veți vedea că am nevoie pentru a crea o variabilă suplimentară topor. Cum pot adauga o legenda graficului meu without a crea această variabilă suplimentară și păstrând simplitatea de actualul meu script.

Games Brainiac
Întrebarea editată 1 octombrie 2013 в 9:12
Programare
python plot matplotlib
1 octombrie 2013 в 8:53
1 vizualizare
Comentarii la întrebare (3)
Soluția

Adauga un label ' = la fiecare plot() telefoane, și apoi apel legenda(loc='din stânga sus').

Luați în considerare această probă (testat cu Python 3.8.0):

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 20, 1000)
y1 = np.sin(x)
y2 = np.cos(x)

plt.plot(x, y1, "-b", label="sine")
plt.plot(x, y2, "-r", label="cosine")
plt.legend(loc="upper left")
plt.ylim(-1.5, 2.0)
plt.show()

Ușor modificate de la acest tutorial: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html

Robᵩ
Răspuns editat 20 noiembrie 2019 в 7:12
Comentarii (5)

Puteți accesa Axe instanță (ax) cu plt.gca(). În acest caz, puteți folosi

plt.gca().legend()

Puteți face acest lucru fie prin utilizarea label ' = de cuvinte cheie în fiecare din plt.complot() apeluri sau prin atribuirea de etichete ca un tuplu sau o listă în "legendă", ca în acest exemplu de lucru:

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-0.75,1,100)
y0 = np.exp(2 + 3*x - 7*x**3)
y1 = 7-4*np.sin(4*x)
plt.plot(x,y0,x,y1)
plt.gca().legend(('y0','y1'))
plt.show()

Cu toate acestea, dacă aveți nevoie pentru a accesa Axe de exemplu mai mult de o dată, eu nu recomand salvarea la variabila ax cu

ax = plt.gca()

și atunci sun ax "în loc de" plt.gca().

cameronroytaylor
Răspuns editat 26 iunie 2018 в 6:12
Comentarii (1)

Aici's un exemplu care să te ajute ...

fig = plt.figure(figsize=(10,5))
ax = fig.add_subplot(111)
ax.set_title('ADR vs Rating (CS:GO)')
ax.scatter(x=data[:,0],y=data[:,1],label='Data')
plt.plot(data[:,0], m*data[:,0] + b,color='red',label='Our Fitting 
Line')
ax.set_xlabel('ADR')
ax.set_ylabel('Rating')
ax.legend(loc='best')
plt.show()

Comentarii (0)

Un simplu complot pentru sinus și cosinus curbe cu o legendă.

Folosit matplotlib.pyplot

import math
import matplotlib.pyplot as plt
x=[]
for i in range(-314,314):
    x.append(i/100)
ysin=[math.sin(i) for i in x]
ycos=[math.cos(i) for i in x]
plt.plot(x,ysin,label='sin(x)')  #specify label for the corresponding curve
plt.plot(x,ycos,label='cos(x)')
plt.xticks([-3.14,-1.57,0,1.57,3.14],['-$\pi$','-$\pi$/2',0,'$\pi$/2','$\pi$'])
plt.legend()
plt.show()

eyllanesc
Răspuns editat 3 septembrie 2018 в 5:45
Comentarii (0)

Adăugați etichete la fiecare argument în teren call corespunzătoare seriei grafice, și anume label ' = "seria 1"

Apoi pur și simplu adăugați Pyplot.legenda() la partea de jos a script-ul dvs. și legenda va afișa aceste etichete.

blaklaybul
Răspuns editat 1 octombrie 2013 в 9:05
Comentarii (1)

Puteți adăuga o legendă personalizate documentation

prima = [1, 2, 4, 5, 4] a doua = [3, 4, 2, 2, 3] plt.complot(în primul rând,'g - ', a doua, 'r,') plt.legenda(['Prima Listă','a Doua Listă'], loc='din stânga sus') plt.show()

Comentarii (0)