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
. Как я могу добавить легенду к моему графику, не создавая эту дополнительную переменную и сохранив простоту моего текущего сценария?
источник
matplotlib
раньше, я хотел сделать вещи максимально простыми. Кроме того, если вы посмотрите на ответ Роба, он будет намного проще, чем пример, показанный на сайте. Надеюсь, это поможет.import matplotlib.pyplot as plt
вместоPyPlot
Ответы:
Добавьте
label=
к каждому из вашихplot()
звонков, а затем позвонитеlegend(loc='upper left')
.Рассмотрим этот пример (протестирован с Python 3.8.0):
Немного изменено из этого урока: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html
источник
plt.legend(loc='upper left')
также работает, гдеplt
находится сimport matplotlib.pyplot as plt
.plt.legend
:plt.legend(['First Label', 'Second Label'])
Вы можете получить доступ к экземпляру Axes (
ax
) с помощьюplt.gca()
. В этом случае вы можете использоватьВы можете сделать это, используя
label=
ключевое слово в каждом из вашихplt.plot()
вызовов или назначив свои метки в виде кортежа или списка внутриlegend
, как в этом рабочем примере:Однако, если вам нужно получить доступ к экземпляру Axes более одного раза, я рекомендую сохранить его в переменной
ax
са потом звоню
ax
вместоplt.gca()
.источник
Вот пример, чтобы помочь вам ...
источник
Простой сюжет для кривых синуса и косинуса с легендой.
Используемый
matplotlib.pyplot
источник
Добавьте метки к каждому аргументу в вашем вызове сюжета, соответствующему серии, которую он отображает, т.е.
label = "series 1"
Затем просто добавьте
Pyplot.legend()
в конец вашего скрипта, и легенда будет отображать эти метки.источник
Вы можете добавить собственную документацию легенды
источник
источник