In [1]:

    

from flockers.model import BoidModel
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline


    

In [2]:

    

def draw_boids(model):
    x_vals = []
    y_vals = []
    for boid in model.schedule.agents:
        x, y = boid.pos
        x_vals.append(x)
        y_vals.append(y)
    fig = plt.figure(figsize=(10,10))
    ax = fig.add_subplot(111)
    ax.scatter(x_vals, y_vals)


    

In [3]:

    

model = BoidModel(100, 100, 100, speed=5, vision=5, separation=1)


    

In [6]:

    

for i in range(50):
    model.step()


    

In [7]:

    

draw_boids(model)


    

In [ ]: