In [7]:

    

%matplotlib inline
import numpy as np
from matplotlib import pyplot as plt


    

In [17]:

    

extent = 2
resolution = 1024
x = np.linspace(-extent,extent,resolution)

X, J = np.meshgrid(x,x)
C = X + J*1j


    

In [20]:

    

def mandelbrot(C,n_iter=40):
    Z = np.zeros_like(C)
    for i in range(n_iter):
        Z = Z**2 + C
    return Z

o = mandelbrot(C)
plt.figure(figsize=(10,10))
plt.imshow(np.abs(o),cmap='gray',extent=[-extent,extent,-extent,extent])


    

    
Out[20]:





<matplotlib.image.AxesImage at 0x7ffb59bb07d0>