Dodanie legendy do PyPlot w Matplotlib w najprostszy możliwy sposób

263

TL; DR -> Jak można stworzyć legendę dla wykresu liniowego w Matplotlib„s PyPlotnie tworząc żadnych dodatkowych zmiennych?

Proszę rozważyć poniższy skrypt do tworzenia wykresów:

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

Jak widać, jest to bardzo proste użycie matplotlib„s PyPlot. To idealnie generuje wykres podobny do tego poniżej:

Wykres

Nic specjalnego, wiem. Nie jest jednak jasne, gdzie są wykreślane dane (próbuję wykreślić dane niektórych algorytmów sortowania, długość w stosunku do czasu i chciałbym się upewnić, że ludzie wiedzą, która linia jest która). Potrzebuję jednak legendy, patrząc na poniższy przykład ( z oficjalnej strony ):

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)

Zobaczysz, że muszę utworzyć dodatkową zmienną ax. Jak mogę dodać legendę do mojego wykresu bez konieczności tworzenia tej dodatkowej zmiennej i zachowując prostotę mojego bieżącego skryptu?

Games Brainiac
źródło
Niepokoi mnie twoja troska o stworzenie dodatkowej zmiennej. W każdym razie musisz robić te obiekty za kulisami.
tacaswell
1
@tcaswell Cóż, pozwól mi spróbować je uspokoić. Nie chcę tworzyć dodatkowych zmiennych, ponieważ zwiększa to złożoność całego skryptu. Staram się tego uczyć wielu studentów, a ponieważ nie korzystali matplotlibwcześniej, chciałem, aby wszystko było jak najprostsze. Ponadto, jeśli spojrzysz na odpowiedź Roba, jest ona o wiele prostsza niż przykład pokazany na stronie internetowej. Mam nadzieję że to pomogło.
Games Brainiac
1
Twierdziłbym, że używanie interfejsu automatu stanów utrudnia zrozumienie na dłuższą metę, ponieważ tak wiele z tego robi się „magią”. Ponadto konwencją jest użycie import matplotlib.pyplot as pltzamiastPyPlot
tacaswell

Odpowiedzi:

441

Dodaj a label=do każdego plot()połączenia, a następnie zadzwoń legend(loc='upper left').

Rozważ tę próbkę (przetestowaną w Pythonie 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()

wprowadź opis zdjęcia tutaj Lekko zmodyfikowany w tym samouczku: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html

Robᵩ
źródło
2
Czy istnieje sposób, aby to zrobić, jeśli nie znasz etykiet w czasie kreślenia serii? Czy to sposób na dodanie etykiet do serii po jej wydrukowaniu? A może sposób na zmodyfikowanie etykiet symboli zastępczych przed wyświetleniem legendy?
davidA,
13
plt.legend(loc='upper left')działa również, skąd pltpochodzi import matplotlib.pyplot as plt.
Matt Kleinsmith,
Dzięki, @eric, za zwrócenie na to uwagi. Zaktualizowałem kod.
Robᵩ
Pozytywnie oceniany, ponieważ przepełnienie stosu jest niesamowite, a dobre odpowiedzi mogą się zmienić
er.
1
@davidA Tak, możesz po prostu przekazać listę ciągów znaków do plt.legend:plt.legend(['First Label', 'Second Label'])
Apollys obsługuje Monikę
36

Możesz uzyskać dostęp do wystąpienia Axes ( ax) za pomocą plt.gca(). W takim przypadku możesz użyć

plt.gca().legend()

Możesz to zrobić, używając label=słowa kluczowego w każdym plt.plot()połączeniu lub przypisując etykiety jako krotkę lub listę wewnątrz legend, jak w tym przykładzie roboczym:

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

pltGcaLegend

Jeśli jednak chcesz uzyskać dostęp do instancji Axes więcej niż raz, zalecamy zapisanie jej w zmiennej za axpomocą

ax = plt.gca()

a następnie dzwonienie axzamiast plt.gca().

cameronroytaylor
źródło
odpowiedź kopiuj-wklej, która nie wymaga czytania, i ze zdjęciem! ta odpowiedź zasługuje na większe uznanie
Gulzar,
14

Oto przykład, który pomoże ci ...

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

wprowadź opis zdjęcia tutaj

Akash Kandpal
źródło
2
Jestem tylko ciekawy, dlaczego twoja linia dopasowania jest tak daleko od danych?
Apollys obsługuje Monikę
13

Prosty wykres krzywych sinus i cosinus z legendą.

Używany 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()

Wykresy grzechu i cosinusa (kliknij, aby zobaczyć obraz)

sajalagrawal
źródło
6

Dodaj etykiety do każdego argumentu w wywołaniu wykresu odpowiadającego serii, na której jest on wyświetlany, tj label = "series 1"

Następnie po prostu dodaj Pyplot.legend()do dolnej części skryptu, a legenda wyświetli te etykiety.

blaklaybul
źródło
To dobry pomysł, ale nigdy nie dodajesz etykiet, więc legenda będzie pusta
tacaswell
4

Możesz dodać niestandardową dokumentację legendy

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

wprowadź opis zdjęcia tutaj

Boris Yakubchik
źródło
0
    # 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()
James Turner
źródło