Commitc449ee4

authored

Merge pull request#16225 from anntzer/animex

Cleanup animation examples.

2 parentsf857586 +d5e8182 commitc449ee4Copy full SHA for c449ee4

File tree

-26

lines changed

-26

lines changed

Lines changed: 4 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -8,13 +8,15 @@`
`8`	`8`	`- changing axes limits during an animation.`
`9`	`9`	`"""`
`10`	`10`
	`11`	`+importitertools`
	`12`	`+`
`11`	`13`	`importnumpyasnp`
`12`	`14`	`importmatplotlib.pyplotasplt`
`13`	`15`	`importmatplotlib.animationasanimation`
`14`	`16`
`15`	`17`
`16`	`18`	`defdata_gen():`
`17`		`-forcntinrange(1000):`
	`19`	`+forcntinitertools.count():`
`18`	`20`	`t=cnt/10`
`19`	`21`	`yieldt,np.sin(2np.pit)*np.exp(-t/10.)`
`20`	`22`
`@@ -47,6 +49,5 @@ def run(data):`
`47`	`49`
`48`	`50`	`returnline,`
`49`	`51`
`50`		`-ani=animation.FuncAnimation(fig,run,data_gen,blit=False,interval=10,`
`51`		`-repeat=False,init_func=init)`
	`52`	`+ani=animation.FuncAnimation(fig,run,data_gen,interval=10,init_func=init)`
`52`	`53`	`plt.show()`

Lines changed: 1 addition & 9 deletions

Original file line number	Diff line number	Diff line change
`@@ -38,16 +38,9 @@ def __init__(self, ax, prob=0.5):`
`38`	`38`	`# which the plotted distribution should converge.`
`39`	`39`	`self.ax.axvline(prob,linestyle='--',color='black')`
`40`	`40`
`41`		`-definit(self):`
`42`		`-self.success=0`
`43`		`-self.line.set_data([], [])`
`44`		`-returnself.line,`
`45`		`-`
`46`	`41`	`def__call__(self,i):`
`47`	`42`	`# This way the plot can continuously run and we just keep`
`48`	`43`	`# watching new realizations of the process`
`49`		`-ifi==0:`
`50`		`-returnself.init()`
`51`	`44`
`52`	`45`	`# Choose success based on exceed a threshold with a uniform pick`
`53`	`46`	`ifnp.random.rand(1,)<self.prob:`
`@@ -62,6 +55,5 @@ def __call__(self, i):`
`62`	`55`
`63`	`56`	`fig,ax=plt.subplots()`
`64`	`57`	`ud=UpdateDist(ax,prob=0.7)`
`65`		`-anim=FuncAnimation(fig,ud,frames=np.arange(100),init_func=ud.init,`
`66`		`-interval=100,blit=True)`
	`58`	`+anim=FuncAnimation(fig,ud,frames=100,interval=100,blit=True)`
`67`	`59`	`plt.show()`

Lines changed: 2 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -79,12 +79,6 @@ def derivs(state, t):`
`79`	`79`	`time_text=ax.text(0.05,0.9,'',transform=ax.transAxes)`
`80`	`80`
`81`	`81`
`82`		`-definit():`
`83`		`-line.set_data([], [])`
`84`		`-time_text.set_text('')`
`85`		`-returnline,time_text`
`86`		`-`
`87`		`-`
`88`	`82`	`defanimate(i):`
`89`	`83`	`thisx= [0,x1[i],x2[i]]`
`90`	`84`	`thisy= [0,y1[i],y2[i]]`
`@@ -94,6 +88,6 @@ def animate(i):`
`94`	`88`	`returnline,time_text`
`95`	`89`
`96`	`90`
`97`		`-ani=animation.FuncAnimation(fig,animate,range(1,len(y)),`
`98`		`-interval=dt*1000,blit=True,init_func=init)`
	`91`	`+ani=animation.FuncAnimation(`
	`92`	`+fig,animate,len(y),interval=dt*1000,blit=True)`
`99`	`93`	`plt.show()`

Lines changed: 1 addition & 6 deletions

Original file line number	Diff line number	Diff line change
`@@ -15,18 +15,13 @@`
`15`	`15`	`line,=ax.plot(x,np.sin(x))`
`16`	`16`
`17`	`17`
`18`		`-definit():# only required for blitting to give a clean slate.`
`19`		`-line.set_ydata([np.nan]*len(x))`
`20`		`-returnline,`
`21`		`-`
`22`		`-`
`23`	`18`	`defanimate(i):`
`24`	`19`	`line.set_ydata(np.sin(x+i/100))# update the data.`
`25`	`20`	`returnline,`
`26`	`21`
`27`	`22`
`28`	`23`	`ani=animation.FuncAnimation(`
`29`		`-fig,animate,init_func=init,interval=2,blit=True,save_count=50)`
	`24`	`+fig,animate,interval=2,blit=True,save_count=50)`
`30`	`25`
`31`	`26`	`# To save the animation, use e.g.`
`32`	`27`	`#`

Comments

(0)