Commit115772a

committed

use np to solve equation

1 parente014950 commit115772aCopy full SHA for 115772a

File tree

+43

-2

lines changed

+43

-2

lines changed

Lines changed: 43 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,10 +1,16 @@`
`1`	`1`	`fromPILimportImage`
`2`	`2`	`fromPILimportImageSequence`
	`3`	`+importnumpyasnp`
	`4`	`+importmatplotlib.pyplotasplt`
`3`	`5`
`4`	`6`
`5`	`7`	`defmain():`
`6`	`8`	`pic_path="ufo/mandelbrot.gif"`
`7`		`-first_step(pic_path)`
	`9`	`+# first_step(pic_path)`
	`10`	`+# second_step()`
	`11`	`+# plt.imshow(mandelbrot(1000, 1000))`
	`12`	`+# plt.show()`
	`13`	`+numPyVersion()`
`8`	`14`
`9`	`15`
`10`	`16`	`deffirst_step(input):`
`@@ -18,9 +24,44 @@ def first_step(input):`
`18`	`24`	`height=480`
`19`	`25`	`forxinrange(width):`
`20`	`26`	`foryinrange(height):`
`21`		`-print(im.getpixel((x,y)),end=" ")`
	`27`	`+print(im.getpixel((x,y)),end=" ")`
`22`	`28`	`print()`
`23`	`29`
`24`	`30`
	`31`	`+defsecond_step():`
	`32`	`+a=np.arange(15).reshape(3,5)`
	`33`	`+print(a)`
	`34`	`+print(a.shape)`
	`35`	`+print(a.ndim)`
	`36`	`+print(a.dtype.name)`
	`37`	`+print(a.dtype)`
	`38`	`+print(a.itemsize)`
	`39`	`+print(type(a))`
	`40`	`+`
	`41`	`+b=np.array([6,7,8])`
	`42`	`+print(b)`
	`43`	`+print(b.shape)`
	`44`	`+pass`
	`45`	`+`
	`46`	`+`
	`47`	`+defmandelbrot(h,w,maxit=20):`
	`48`	`+y,x=np.ogrid[-1.4:1.4:h1j,-2:0.8:w1j]`
	`49`	`+c=x+y*1j`
	`50`	`+z=c`
	`51`	`+divtime=maxit+np.zeros(z.shape,dtype=int)`
	`52`	`+`
	`53`	`+foriinrange(maxit):`
	`54`	`+z=z**2+c`
	`55`	`+diverge=znp.conj(z)>2*2`
	`56`	`+div_now=diverge& (divtime==maxit)`
	`57`	`+divtime[div_now]=i`
	`58`	`+z[diverge]=2`
	`59`	`+returndivtime`
	`60`	`+`
	`61`	`+`
	`62`	`+defnumPyVersion():`
	`63`	`+print(np.version.version)`
	`64`	`+`
	`65`	`+`
`25`	`66`	`if__name__=="__main__":`
`26`	`67`	`main()`

Comments

(0)