Додавання легенди до PyPlot в Matplotlib найпростішим способом


263

TL; DR -> Як можна створити легенду для лінійного графіка в Matplotlib's PyPlotбез створення зайвих змінних?

Будь ласка, врахуйте графічний сценарій нижче:

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

Як бачите, це дуже основне використання matplotlib's PyPlot. Це в ідеалі генерує графік, як показано нижче:

Графік

Нічого особливого, я знаю. Однак незрозуміло, які дані наводяться де (я намагаюся побудувати дані деяких алгоритмів сортування, тривалості від часу, і я хотів би переконатися, що люди знають, яка лінія). Таким чином, мені потрібна легенда, дивлячись на наступний приклад нижче ( з офіційного сайту ):

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)

Ви побачите, що мені потрібно створити додаткову змінну ax. Як я можу додати в свій графік легенду, не створюючи цю додаткову змінну і не зберігши простоту мого поточного сценарію?


Мене бентежить ваше занепокоєння щодо створення додаткової змінної. Ви повинні робити ці об'єкти за лаштунками в будь-якому випадку.
tacaswell

1
@tcaswell Добре дозвольте спробувати переконати їх. Я не хочу створювати зайві змінні, тому що це додає складності всьому сценарію. Я намагаюсь навчити цього ряду учнів, і оскільки вони раніше не використовувались matplotlib, я хотів зробити все максимально простим. Крім того, якщо ви подивитесь на відповідь Роба, це набагато простіше, ніж приклад, показаний на веб-сайті. Я сподіваюся, що це допомагає.
Ігри Brainiac

1
Я заперечую, що використання інтерфейсу державної машини ускладнює розуміння в довгостроковій перспективі, оскільки стільки цього робиться "за допомогою магії". Також import matplotlib.pyplot as pltPyPlot
умовою

Відповіді:


441

Додайте label=до кожного свого plot()дзвінка, а потім зателефонуйте legend(loc='upper left').

Розглянемо цей зразок (тестований на 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()

введіть тут опис зображення Трохи модифіковані з цього підручника: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html


2
Чи є спосіб це зробити, якщо ви не знаєте етикетки під час побудови серії? Тобто спосіб додати мітки до серії після того, як вона вже була побудована? Чи, можливо, спосіб зміни міток-заповнювачів перед тим, як показати легенду?
davidA

13
plt.legend(loc='upper left')також працює, де pltзнаходиться з import matplotlib.pyplot as plt.
Метт Клейнсміт

Дякую, @eric, що зауважив це. Я оновив код.
Robᵩ

Запропоновано, оскільки переповнення стека є приголомшливим, і хороші відповіді можуть змінитися
eric

1
@davidA Так, ви можете просто передати список рядків plt.legend:plt.legend(['First Label', 'Second Label'])
Apollys підтримує Моніку

36

Ви можете отримати доступ до екземпляра Axes ( ax) за допомогою plt.gca(). У цьому випадку можна використовувати

plt.gca().legend()

Це можна зробити, використовуючи label=ключове слово в кожному з ваших plt.plot()дзвінків, або призначивши свої мітки як кортеж або список у межах legend, як у цьому робочому прикладі:

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

Тим НЕ менше, якщо вам потрібно отримати доступ до примірника Топоров більш , що коли - то, я рекомендую зберегти його в змінний axз

ax = plt.gca()

а потім дзвонити axзамість plt.gca().


скопіюйте-вставте відповідь, яка не потребує читання, та із зображенням! ця відповідь заслуговує більшого кредиту
Gulzar

14

Ось приклад, який допоможе вам ...

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

введіть тут опис зображення


2
Мені просто цікаво, чому ваша відповідна лінія настільки далека від даних?
Аполіс підтримує Моніку

13

Простий сюжет для синусоїдних і косинусних кривих з легендою.

Б / в 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()

Сюжети гріха та косину (натисніть, щоб переглянути зображення)


6

Додайте мітки до кожного аргументу у вашому виклику сюжету, що відповідає серії, на якій він зображений, тобто label = "series 1"

Потім просто додайте Pyplot.legend()внизу вашого сценарію, і легенда відобразить ці мітки.


Це правильна ідея, але ви ніколи не додаєте етикетки, тому легенда буде порожньою
tacaswell


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()
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.