Adicionando uma legenda ao PyPlot no Matplotlib da maneira mais simples possível


263

TL; DR -> Como se pode criar uma legenda para um gráfico de linhas Matplotlib, PyPlotsem criar variáveis ​​extras?

Por favor, considere o script gráfico abaixo:

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

Como você pode ver, este é um uso muito básico de matplotlib's PyPlot. Idealmente, isso gera um gráfico como o abaixo:

Gráfico

Nada de especial, eu sei. No entanto, não está claro quais dados estão sendo plotados onde (estou tentando plotar os dados de alguns algoritmos de classificação, a duração e o tempo gasto e gostaria de garantir que as pessoas saibam qual linha é qual). Portanto, preciso de uma lenda, no entanto, dando uma olhada no exemplo a seguir abaixo ( no site 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)

Você verá que eu preciso criar uma variável extra ax. Como posso adicionar uma legenda ao meu gráfico sem precisar criar essa variável extra e manter a simplicidade do meu script atual?


Estou confuso com sua preocupação em criar uma variável extra. Você tem que fazer esses objetos nos bastidores de qualquer maneira.
tacaswell

1
@tcaswell Bem, deixe-me tentar acalmá-los. Eu não quero criar variáveis ​​extras, porque adiciona complexidade a todo o script. Estou tentando ensinar isso a um monte de estudantes e, como eles não foram usados matplotlibantes, eu queria manter as coisas o mais simples possível. Além disso, se você der uma olhada na resposta de Rob, é muito mais simples que o exemplo mostrado no site. Espero que ajude.
Games Brainiac

1
Eu argumentaria que o uso da interface da máquina de estado torna mais difícil entender a longo prazo, porque muito disso está sendo feito "por mágica". Além disso, a convenção é usar import matplotlib.pyplot as plt, em vez dePyPlot
tacaswell

Respostas:


441

Adicione a label=a cada uma das suas plot()chamadas e, em seguida, ligue legend(loc='upper left').

Considere este exemplo (testado com 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()

insira a descrição da imagem aqui Ligeiramente modificado neste tutorial: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html


2
Existe uma maneira de fazer isso se você não souber os rótulos no momento em que a série é plotada? Ou seja, uma maneira de adicionar etiquetas a uma série depois que ela já foi plotada? Ou talvez uma maneira de modificar os marcadores de espaço reservado antes de mostrar a legenda?
precisa saber é o seguinte

13
plt.legend(loc='upper left')também funciona, de onde plté import matplotlib.pyplot as plt.
precisa saber é o seguinte

Obrigado, @eric, por observar isso. Eu atualizei o código.
Robᵩ

Upvoted porque estouro de pilha é incrível e boas respostas podem mudar
eric

1
@davidA Sim, você pode simplesmente passar uma lista de strings para plt.legend:plt.legend(['First Label', 'Second Label'])
Apollys suporta Monica

36

Você pode acessar a instância do Axes ( ax) com plt.gca(). Nesse caso, você pode usar

plt.gca().legend()

Você pode fazer isso usando a label=palavra - chave em cada uma das suas plt.plot()chamadas ou atribuindo seus marcadores como tupla ou lista legend, como neste exemplo de trabalho:

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

No entanto, se você precisar acessar a instância do Axes mais de uma vez, recomendo salvá-la na variável axcom

ax = plt.gca()

e depois chamando em axvez de plt.gca().


resposta de copiar e colar que não requer leitura e com uma foto! esta resposta merece mais crédito
Gulzar

14

Aqui está um exemplo para ajudá-lo ...

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

insira a descrição da imagem aqui


2
Só estou curioso, por que sua linha de encaixe está tão longe dos dados?
Apollys suporta Monica 12/12/19

13

Um gráfico simples para curvas seno e cosseno com uma lenda.

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

Parcelas Sin e Cosine (clique para ver a imagem)


6

Adicione rótulos a cada argumento em sua chamada de plotagem correspondente à série que está representando graficamente, ou seja, label = "series 1"

Em seguida, basta adicionar Pyplot.legend()na parte inferior do seu script e a legenda exibirá esses rótulos.


Esta é a idéia certa, mas você nunca adicionar os rótulos para a legenda estará vazia
tacaswell

4

Você pode adicionar uma documentação de legenda personalizada

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

insira a descrição da imagem aqui


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()
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.