Как нарисовать легенду в python pyplot?

Question

[Scheuble]

Имеется лист с x/y - координатами точек, которые я рисую в pyplot.
Проходясь for циклом по листу я вызываю plt.plot()

for h in points:
    plt.plot(h[0], h[1], 'k^', label='точка')

plt.legend(loc='upper left', frameon=True)

Получается так, что легенда рисуется ровно столько раз, сколько точек в листе, а нужна всего лишь 1 раз (т.к. точки сами по себе ничем друг от друга не отличаются)
Как это исправить?

Answer 1

[dmshar]

По логике вашего скрипта непонятно, а что же вы хотите в той легенде написать?
Тогда решение "в лоб".
Внесите вызов функции plt.legend в цикл и вызывайте ее только на первом шаге цикла. Ну, или только один раз.

Comments

[Scheuble]

В том то и дело, что вызов функции рисования (plt.legend) происходит только 1 раз, однако легенда появляется к каждой точке, которая рисуется в for-цикле, при этом точки эти не отличаются друг от друга, тоесть и легенда должна рисоваться всего лишь 1 раз. Смотрите график:

[dmshar]

К сожалению, вы так и не поняли, что я вам насоветовал.
Я рекомендовал вам вызывать функцию один раз, в самом начале.
А у вас она вызывается в конце, тогда, когда вы натыкали в ваш рисунок все свои точки.

import matplotlib.pyplot as plt
points=[[1,2],[3,4],[5,6],[7,8],[9,10]]
for i, h in enumerate(points):
    plt.plot(h[0], h[1], 'k^', label='точка')
    if i==0:
        plt.legend(loc='upper left', frameon=True)

Кстати, это конечно не лучший вариант решения, а просто попытка выкрутиться из того решения, которое вы приняли. Более квалифицированное решение выглядит, очевидно, вот так:

import matplotlib.pyplot as plt
points=[[1,2],[3,4],[5,6],[7,8],[9,10]]
hh0=[]
hh1=[]
for h in points:
    hh0.append(h[0])
    hh1.append(h[1])
plt.plot(hh0, hh1, 'bo', label='точка')
plt.legend(loc='upper left', frameon=True)

Картинка тождественна предыдущей.

[Scheuble]

dmshar, большое спасибо за разъяснения.
Я предпологал, что вызов plt.legend вне for-цикла:

import matplotlib.pyplot as plt
points=[[1,2],[3,4],[5,6],[7,8],[9,10]]
for i, h in enumerate(points):
    plt.plot(h[0], h[1], 'k^', label='точка')

plt.legend(loc='upper left', frameon=True)

равнозначно тому, как предложили вы, вызов внутри for-цикла с ограничением if i==0.
Как оказалось, это не одно и тоже. Подскажете почему?

Спасибо!

[dmshar]

Scheuble, Потому что legend - это отдельный объект, связанный с графиком. И каждый вызов метода plot с параметром label "кладет" в этот объект ЕЩЕ одну запись-строчку. Т.е. каждая запись в легенде связана с тем, что рисует plot. А вывод этой легенды - отдельное действие.
Вы в своем цикле для чего-то рисуете на самом деле МНОГО разных графиков, но каждый из одной единственной точки. А потом всем скопом выводите всю накопившуюся легенду. Поэтому я и предложил - либо сразу, после первой точки выводить легенду, либо рисовать один график, с единственной записью в легенде. Ну, можно еще рисовать один график с созданием строчки в легенде, а остальные - без.
Но все это извращение, и потребоваться оно может только в очень специфических случаях. Лучший вариант - и самый быстрый и эффективный по памяти - рисовать один график (второй мой вариант).