Commitde07ea1

committed

Fix use of Heaviside in lambdify

1 parent1079356 commitde07ea1Copy full SHA for de07ea1

File tree

+16

-13

lines changed

+16

-13

lines changed

Lines changed: 16 additions & 13 deletions

Original file line number	Diff line number	Diff line change
`@@ -11,26 +11,29 @@`
`11`	`11`
`12`	`12`	`fromsympyimportSymbol,Heaviside,lambdify`
`13`	`13`
`14`		`-ta= [0,4,8,12,16];tb= [2,6,10,14,18]`
`15`		`-ba=Symbol('ba');bb=Symbol('bb');t=Symbol('t')`
`16`		`-fa=sum([Heaviside(t-_t)for_tinta])*ba`
`17`		`-fb=sum([Heaviside(t-_t)for_tintb])*bb`
`18`		`-N=fa+fb`
	`14`	`+ta= [0,4,8,12,16]`
	`15`	`+tb= [2,6,10,14,18]`
	`16`	`+ba=Symbol('ba')`
	`17`	`+bb=Symbol('bb')`
	`18`	`+t=Symbol('t')`
	`19`	`+fa=sum([Heaviside(t-_t)for_tinta])*ba`
	`20`	`+fb=sum([Heaviside(t-_t)for_tintb])*bb`
	`21`	`+N=fa+fb`
`19`	`22`
`20`		`-Nn=N.subs(ba,1)`
`21`		`-Nn=Nn.subs(bb,-2)`
	`23`	`+Nn=N.subs(ba,1)`
	`24`	`+Nn=Nn.subs(bb,-2)`
`22`	`25`
`23`		`-Nn=lambdify(t,Nn)`
	`26`	`+# Use Numpy heaviside for lambdify, with y=1 for x=0.`
	`27`	`+modules= [{'Heaviside':lambdax:np.heaviside(x,1)},'numpy']`
	`28`	`+neuronal_func=lambdify(t,Nn,modules=modules)`
`24`	`29`
`25`		`-tt=np.linspace(-1,21,1201)`
`26`		`-neuronal= [Nn(_t)for_tintt]`
`27`		`-# Deal with undefined Heaviside at 0`
`28`		`-neuronal= [n.subs(Heaviside(0.0),1)forninneuronal]`
	`30`	`+tt=np.linspace(-1,21,1201)`
	`31`	`+neuronal=neuronal_func(tt)`
`29`	`32`
`30`	`33`	`plt.step(tt,neuronal)`
`31`	`34`
`32`	`35`	`a=plt.gca()`
`33`		`-a.set_ylim([-5.5,1.5])`
	`36`	`+a.set_ylim([-5.5,1.5])`
`34`	`37`	`a.set_ylabel('Neuronal (cumulative)')`
`35`	`38`	`a.set_xlabel('Time')`
`36`	`39`

Comments

(0)