Как добавить легенду / метку в анимации python

Я вообще не специалист по matplotlib, но в анимации Double Pendulum они отображают тексты, которые меняются, и это приводит к некоторым вариациям, которые могут вам помочь.

Чтобы получить легенды с фактическим цветом линий, вы можете либо изменить начальные установочные lines :

 lines = [plt.plot([], [], label = 'line {}'.format(i))[0] for i in range(N)] 

или добавьте line.set_label() в цикл for в функции init() . Оба они, похоже, работают так, как ожидалось. По крайней мере, если вы добавите plt.legend(loc="upper left") прямо перед plt.show() .

Однако set_label не работает в функции set_label animate() , но согласно связанной анимации вы можете использовать определенные текстовые поля, добавленные в анимацию, и это, похоже, работает хорошо. Добавьте следующий код после инициализации lines :

 texts = [ax.text(0.80, 0.95-i*0.05, '', transform=ax.transAxes) for i in range(N)] 

И измените animate() следующим образом:

 def animate(i): for j in range(N): lines[j].set_data([0, 2], [10*j,i]) #, label="hei {}".format(i)) texts[j].set_text('line %d, stage %d'%(j,i)) return lines 

Это помещает текст в верхний правый угол и обновляется для каждого шага анимации. Так как строки все еще имеют свою легенду, вы можете упростить в один текст, отображающий только сцену. Но я оставляю тонкую настройку сообщений на ваше усмотрение.

Приложение: Расширение Line2D

Другой альтернативой может быть расширение lines.Line2D и использование этих строк в вашей анимации, что-то похожее на эту статью . Не уверен, что это будет работать с анимацией, но если вы не можете заставить выше работать, это может стоить попробовать.

вы должны вернуть легенду в своей анимационной функции, чтобы она была визуализирована. Попробуйте это, вместо этого:

 legend = plt.legend() def animate(i): for j,line in enumerate(lines): line.set_data([0, 2], [10*j,i]) line.set_label('line %d, stage %d'%(j,i)) legend.remove() legend = plt.legend() return lines + [legend] 

Вы также должны включить тот же код в свою функцию init, init используется при изменении размера окна, иначе легенда исчезнет при изменении размера