sierpinski.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jan 22 16:51:30 2019

@author: toabi
"""

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.animation as animation

choice = np.array([0,0,0,0,0,0,0,0,0,0,0,0,0,0,1])

C = np.array([[0,0,0,0,0,0,0],
             [0,0,0,0,0,0,0],
             [0,0,0,1,0,0,0],
             [0,0,0,0,0,0,0],
             [0,0,0,0,0,0,0],
             [0,0,0,0,0,0,0]])

A = np.random.randint(2, size=(50, 50))
#Z = np.random.choice(choice,size = (200,200))

Z = np.zeros((200,200))
Z[100,100] = 1
#Z[99,99] = 1
print(Z)

def animate(i):
    def iterate(Z):
        # Count neighbours
        N = (Z[0:-2,0:-2] + Z[0:-2,1:-1] + Z[0:-2,2:] +
             Z[1:-1,0:-2]                + Z[1:-1,2:] +
             Z[2:  ,0:-2] + Z[2:  ,1:-1] + Z[2:  ,2:])
        
        #rules
        birth = (N==1) & (Z[1:-1,1:-1]==0)
        survive = ((N==1) | (N==2)) & (Z[1:-1,1:-1]==1)
        Z[...] = 0
        Z[1:-1,1:-1][birth | survive] = 1
        return Z

    
    iterate(Z)
    p.set_data(Z)
       
    
fig = plt.figure()
p = plt.imshow(Z,interpolation='nearest', animated = True)
plt.set_cmap('rainbow')

ani = animation.FuncAnimation(fig, animate, interval=1)
plt.show()