import time import numpy as np def mandelbrot(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 tic = time.perf_counter_ns() image = mandelbrot() toc = time.perf_counter_ns() print((toc - tic)/1_000_000_000, "s")