Добавление легенды в PyPlot в Matplotlib самым простым способом

TL;DR -> Как можно создать легенду для линейного графика в Matplotlib'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'а 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. Как я могу добавить легенду к моему графику без необходимости создавать эту дополнительную переменную и сохранить простоту моего текущего скрипта.

Games Brainiac
Редактировал вопрос 1-го октября 2013 в 9:12
Программирование
python plot matplotlib
1-го октября 2013 в 8:53
24 просмотров
Комментарии к вопросу (3)
Решение

Добавьте 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

Robᵩ
Редактировал ответ 20-го ноября 2019 в 7:12
Комментарии (5)

Вы можете получить доступ к экземпляру 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()

Однако, если вам нужно получить доступ к экземпляру Axes более одного раза, я рекомендую сохранить его для переменной ax с

ax = plt.gca()

и затем вызывает ax вместо plt.gca ().

cameronroytaylor
Редактировал ответ 26-го июня 2018 в 6:12
Комментарии (1)

Вот пример, который вам поможет...

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

Комментарии (0)

Простой сюжет для синусоидальных и косинусных кривых с легендой.

Используется 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()

Площадь и Cosine графиков (нажмите, чтобы посмотреть изображение)!

eyllanesc
Редактировал ответ 3-го сентября 2018 в 5:45
Комментарии (0)

Добавьте метки к каждому аргументу в вызове plot, соответствующие серии, которую он отображает, например label = "series 1".

Затем просто добавьте Pyplot.legend() в нижнюю часть вашего скрипта, и легенда будет отображать эти метки.

blaklaybul
Редактировал ответ 1-го октября 2013 в 9:05
Комментарии (1)

Вы можете добавить пользовательскую легенду документация

первый = [1, 2, 4, 5, 4]
второй = [3, 4, 2, 2, 3]
plt.plot (первый, 'g--', второй, 'r--')
plt.legend ([«Первый список», «Второй список»], loc = «верхний левый»)
plt.show ()

Комментарии (0)