Я хочу добавить легенду в анимацию Python, какline.set_label() ниже. Это похоже наplt.plot(x,y,label='%d' %*variable*).

Тем не менее, я считаю, что коды не работают здесь. Анимация показывает только изменяющиеся линии, но нет метки или легенды. Как я могу решить эту проблему?

 from matplotlib import pyplot as plt 
 from matplotlib import animation

fig = plt.figure()

ax = plt.axes(xlim=(0, 2), ylim=(0, 100))

N = 3
lines = [plt.plot([], [])[0] for _ in range(N)]


def init():    
    for line in lines:
        line.set_data([], [])
    return lines

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))
    return lines

anim = animation.FuncAnimation(fig, animate, init_func=init,
                               frames=100, interval=20, blit=True)

plt.show()