Python动画阴影matplotlib

Question

所以我试图创建一些随时间推移的高斯图的动画。我的基本动画也按我的意图工作。但后来我想遮住高斯曲线下的区域。 我用ax.fill函数来遮挡它。但是，它只是阴影创建曲线的绘图矢量，它只是变得一团糟。任何建议或帮助将不胜感激

import numpy as np 
import matplotlib.pyplot as plt 
plt.switch_backend('agg') 
import matplotlib.animation as animation 
import pylab as p 

Gamma=0.0005 
q=1.6e-19 
m=0.067*9e-31 
B=10 
Ec=(1.0567e-34)*B/m 
#e=2.78 

#E0=0+(1.0567e-34)*x*i/m 

fig, ax = plt.subplots() 

n = 3 #number of lines 
x = np.arange(0, 3.6e-3, 1.7e-5)  # x-array, third number is interval here, x is energy 
lines = [ax.plot(x, np.e**(-(x-((1.0567e-34)*1*1/m))**2/Gamma**2))[0]for _ in range(n)] 


def animate(i): 
    for d, line in enumerate(lines): 
     p=(d+1)/2. 
     line.set_ydata(np.e**((-(x-((1.0567e-34)*p*i/m))**2)/Gamma**2)) 
     ax.fill(x,(np.e**(-(x-((1.0567e-34)*p*i/m))**2/Gamma**2)), "b")# update the data 
    return lines 


#Init only required for blitting to give a clean slate. 
def init(): 
    for line in lines: 
     line.set_ydata(np.ma.array(x, mask=True)) 
    return lines 

ani = animation.FuncAnimation(fig, animate, np.arange(0, 2.5, .01), init_func=init, 
    interval=10, blit=True) 
Writer = animation.writers['ffmpeg'] 
writer = Writer(fps=20, metadata=dict(artist='Me'), bitrate=1800) 

ani.save('QHanimati.mp4', writer=writer) 



plt.show() 

所以当前的代码运行一个动画，给出这样的事情。 Curve

而且我想让它离开白色曲线上方的区域。

谢谢。

Answer 1

您需要知道，保存的动画是如果blit=False将显示的动画。所以会发生什么是填充全部添加到画布上。

一个解决方案是删除前一个之前添加下一个（因为像set_data不直接存在填充）。

再次，我们将创建一个填充列表，实际使用fill_between而不是fill，我们可以使用它来删除填充并向其中添加新填充。

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


Gamma=0.0005 
q=1.6e-19 
m=0.067*9e-31 
B=10 
Ec=(1.0567e-34)*B/m 

fig, ax = plt.subplots() 

n = 3 #number of lines 
x = np.arange(0, 3.6e-3, 1.7e-5)  # x-array, third number is interval here, x is energy 
lines = [ax.plot(x, np.e**(-(x-((1.0567e-34)*1*1/m))**2/Gamma**2), zorder=i+3)[0] for i in range(n)] 
fills = [ax.fill_between(x,0,(np.e**(-(x-((1.0567e-34)*1*1/m))**2/Gamma**2)), facecolor=lines[i].get_color(), zorder=i+3) for i in range(n)] 


def animate(i): 
    for d, line in enumerate(lines): 
     p=(d+1)/2. 
     line.set_ydata(np.e**((-(x-((1.0567e-34)*p*i/m))**2)/Gamma**2)) 
     fills[d].remove() 
     fills[d] = ax.fill_between(x,0,(np.e**(-(x-((1.0567e-34)*p*i/m))**2/Gamma**2)), facecolor=lines[d].get_color(), zorder=d+3)# update the data 

    return lines + fills 


#Init only required for blitting to give a clean slate. 
def init(): 
    for line in lines: 
     line.set_ydata(np.ma.array(x, mask=True)) 
    return lines 

ani = animation.FuncAnimation(fig, animate, np.arange(0, 2.5, .01), init_func=init, 
    interval=10, blit=True) 
#Writer = animation.writers['ffmpeg'] 
#writer = Writer(fps=20, metadata=dict(artist='Me'), bitrate=1800) 
# 
#ani.save('QHanimati.mp4', writer=writer) 

plt.show()