Mandelbrot.py

import numpy as np
import matplotlib.pyplot as plt

class Mandelbrot(object):

    def __init__(self):
        n = 500
        self.n = n
        self.grid = np.zeros((n,n))
        x = np.linspace(-2.025, 0.6, n)
        y = np.linspace(-1.125, 1.125, n)
        self.xx, self.yy = np.meshgrid(x,y)

    def establishComplex(self, Re, Im):
        c = complex(Re, Im)
        z = 0.0j
        maxiter = 255
        for i in range(maxiter):
            z = z * z + c
            if abs(z) >= 2:
                return i
        return maxiter

    def generate(self):
        for i in range(self.n):
            for j in range(self.n):
                    Re = self.xx[j,i]
                    Im = self.yy[j,i]
                    result = self.establishComplex(Re, Im)
                    self.grid[i,j] = result

    def plotSet(self):
        plt.figure(dpi = 100)
        plt.imshow(self.grid.T,cmap='hot',interpolation='none',extent=[-2.025, 0.6, -1.125, 1.125])
        plt.title('mandelbrot set')
        plt.xlabel('Re')
        plt.ylabel('Im')
        plt.show

def main():
    m = Mandelbrot()
    m.generate()
    m.plotSet()

main()