推荐到我的博客享受更好的显示
在上一篇文章中,我们已经完成了干涉图样的绘制,现在我们希望能直观地了解干涉图样随d取值的变化而发生的改变,这里我们需要调用animation.ArtistAnimation函数,这个函数的功能是根据我们已有的一组图片,自动地生成对应的gif文件。
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import animation
import math
def update(i):
xmax = 20
Lambda = 632.8*10**(-6)
f = 200
n = 1.0
N = 300
d = 0.3+i
x,y= np.meshgrid(np.linspace(-xmax,xmax,N),np.linspace(-xmax,xmax,N))
r = np.sqrt(x**2+y**2)
B=np.cos(math.pi*(2*n*d*np.cos(np.arcsin(n*np.sin(np.arctan(r/f)))))/Lambda)**2
im=plt.imshow(B).findobj()
ims.append(im)
if __name__ == "__main__": #虽然不知道为什么但大家都这么写
fig = plt.figure() #创建一个新的窗口
ims = [] #创建一个列表,之后会在列表中填充每一帧的动画
for i in np.linspace(0,0.001,100): #让i在0到0.001之间循环
update(i) 调用函数
ani = animation.ArtistAnimation(fig, ims, interval=100, repeat_delay=100) #通过Artist Animation创建动图
ani.save("test.gif",writer='pillow') #将动图保存到test.gif
等到的结果是: