diamoeba.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jan 22 17:25:34 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])
choice2 = np.array([0,1])
prob = [0.52,0.48]

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]])

Z = np.random.randint(2, size=(300, 300))
#Z = np.random.choice(choice2,size = (300,300),p = prob)

#Z = np.zeros((100,100))
#Z[1,1] = 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==3) | (N==5) | (N==6) | (N==7) | (N==8)) & (Z[1:-1,1:-1]==0)
        survive = ((N==5) | (N==6) | (N==7) | (N==8)) & (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('winter')

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