25 lines
594 B
Python
25 lines
594 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
def mandelbrot_np(
|
||
|
|
pmin=-2.5,
|
||
|
|
pmax=1.5,
|
||
|
|
qmin=-2,
|
||
|
|
qmax=2,
|
||
|
|
ppoints=200,
|
||
|
|
qpoints=200,
|
||
|
|
max_iterations=300,
|
||
|
|
infinity_border=100):
|
||
|
|
image = np.zeros((ppoints, qpoints))
|
||
|
|
|
||
|
|
for ip, p in enumerate(np.linspace(pmin, pmax, ppoints)):
|
||
|
|
for iq, q in enumerate(np.linspace(qmin, qmax, qpoints)):
|
||
|
|
c = p + 1j * q
|
||
|
|
z = 0
|
||
|
|
for k in range(max_iterations):
|
||
|
|
z = z ** 2 + c
|
||
|
|
if abs(z) > infinity_border:
|
||
|
|
|
||
|
|
image[ip, iq] = 1
|
||
|
|
break
|
||
|
|
return image
|