如何在matplotlib动画中更新标题？

问题描述

我正在尝试将标题设置为“模式（i）的电场”，其中我从1到N，但是我找不到路。导入Numpy，Matplotlib，cmath和Animation，并定义参数和必要的功能（模）之后，我有：

fig = plt.figure()
subplot = plt.axes(xlim=(0,60*(10**(-15))),xlabel=("t[s]"),ylim=(-10,10),ylabel=("Amplitud [u.a.]"))
modo,= subplot.plot([],[],lw=2)
    def init1():
     modo.set_data([],[])
     return modo,def animacion1(i):
     EiR,Ei = Modo(Eo,wo,dw,t,phio,i)
     modo.set_data(t,EiR)
     return modo,anim1 = animation.FuncAnimation(fig,animacion1,init_func=init1,frames=N,interval=20,blit=True,save_count=20000)
plt.show()

解决方法

这是一个最小的示例：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation


x = np.linspace(0,7.5,100)
y1 = np.sin(x)

fig,ax = plt.subplots()

line1,= ax.plot(x,y1,color='orange')

def update(num):
    line1.set_data(x[:num],y1[:num])
    ax.set_title(f'{num}')
    return line1

ani = animation.FuncAnimation(fig,func=update,frames=len(x),interval = 50)

plt.show()

根据您的情况，您需要使用subplot.set_title(f'Electric Field of mode ({i})')方法进行animacion1()。