Aggiunta di una legenda a PyPlot in Matplotlib nel modo più semplice possibile

TL; DR -> Come si può creare una legenda per un grafico a linee in Matplotlibs' PyPlotsenza creare variabili aggiuntive?

Si prega di considerare lo script grafico qui sotto:

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

Come si può vedere, si tratta di un uso di base di matplotlibs' PyPlot. Questo genera idealmente un grafico come quello qui sotto:

Niente di speciale, lo so. Tuttavia, non è chiaro quali dati vengano tracciati dove (sto cercando di tracciare i dati di alcuni algoritmi di ordinamento, la lunghezza rispetto al tempo impiegato e vorrei assicurarmi che le persone sapessero quale linea fosse quale). Pertanto, ho bisogno di una legenda, tuttavia, dando un'occhiata al seguente esempio di seguito ( dal sito ufficiale ):

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)

Vedrai che devo creare una variabile aggiuntiva ax. Come posso aggiungere una legenda al mio grafico senza dover creare questa variabile aggiuntiva e conservare la semplicità del mio script attuale?

Aggiungi label=a ciascuna delle tue plot()chiamate, quindi chiama legend(loc='upper left').

Considera questo esempio (testato con 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()

Leggermente modificato da questo tutorial: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html

È possibile accedere all'istanza Axes ( ax) con plt.gca(). In questo caso, puoi usare

plt.gca().legend()

Puoi farlo utilizzando la label=parola chiave in ciascuna delle tue plt.plot()chiamate o assegnando le etichette come tupla o elenco all'interno legend, come in questo esempio funzionante:

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

Tuttavia, se è necessario accedere all'istanza Axes più di una volta, si consiglia di salvarlo nella variabile axcon

ax = plt.gca()

e quindi chiamare axinvece di plt.gca().

Ecco un esempio per aiutarti ...

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

Una trama semplice per curve seno e coseno con una leggenda.

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

Aggiungi etichette a ciascun argomento nella tua chiamata alla trama corrispondente alla serie che sta rappresentando, ad es label = "series 1"

Quindi aggiungi semplicemente Pyplot.legend()in fondo allo script e la legenda visualizzerà queste etichette.

Puoi aggiungere una legenda personalizzata documentazione della

first = [1, 2, 4, 5, 4]
second = [3, 4, 2, 2, 3]
plt.plot(first,'g--', second, 'r--')
plt.legend(['First List','Second List'], loc='upper left')
plt.show()

    # Dependencies
    import numpy as np
    import matplotlib.pyplot as plt

    #Set Axes
    # Set x axis to numerical value for month
    x_axis_data = np.arange(1,13,1)
    x_axis_data

    # Average weather temp
    points = [39, 42, 51, 62, 72, 82, 86, 84, 77, 65, 55, 44]

    # Plot the line
    plt.plot(x_axis_data, points)
    plt.show()

    # Convert to Celsius C = (F-32) * 0.56
    points_C = [round((x-32) * 0.56,2) for x in points]
    points_C

    # Plot using Celsius
    plt.plot(x_axis_data, points_C)
    plt.show()

    # Plot both on the same chart
    plt.plot(x_axis_data, points)
    plt.plot(x_axis_data, points_C)

    #Line colors
    plt.plot(x_axis_data, points, "-b", label="F")
    plt.plot(x_axis_data, points_C, "-r", label="C")

    #locate legend
    plt.legend(loc="upper left")
    plt.show()