TL; DR -> Jak można stworzyć legendę dla wykresu liniowego w
Matplotlib
„sPyPlot
nie 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:
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?
źródło
matplotlib
wcześ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.import matplotlib.pyplot as plt
zamiastPyPlot
Odpowiedzi:
Dodaj a
label=
do każdegoplot()
połączenia, a następnie zadzwońlegend(loc='upper left')
.Rozważ tę próbkę (przetestowaną w Pythonie 3.8.0):
Lekko zmodyfikowany w tym samouczku: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html
źródło
plt.legend(loc='upper left')
działa również, skądplt
pochodziimport matplotlib.pyplot as plt
.plt.legend
:plt.legend(['First Label', 'Second Label'])
Możesz uzyskać dostęp do wystąpienia Axes (
ax
) za pomocąplt.gca()
. W takim przypadku możesz użyćMożesz to zrobić, używając
label=
słowa kluczowego w każdymplt.plot()
połączeniu lub przypisując etykiety jako krotkę lub listę wewnątrzlegend
, jak w tym przykładzie roboczym:Jeśli jednak chcesz uzyskać dostęp do instancji Axes więcej niż raz, zalecamy zapisanie jej w zmiennej za
ax
pomocąa następnie dzwonienie
ax
zamiastplt.gca()
.źródło
Oto przykład, który pomoże ci ...
źródło
Prosty wykres krzywych sinus i cosinus z legendą.
Używany
matplotlib.pyplot
źródło
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.źródło
Możesz dodać niestandardową dokumentację legendy
źródło
źródło