Exemplo n.º 1
0
def simulate_walk(distance, n_iterations, walker_class):
    '''
    Walk simulation

    ## Params:
    - distance, int:

        Number of steps

    - n_iterations, int:

        Number of simulations

    - walker_class, Walker:

        walker class that will be used for simulation

    ## Returns
        List of distances to origin for each walk simulation
    '''
    walker = walker_class()
    origin = Coordinate(0, 0)
    distances = []

    for _ in range(n_iterations):
        system = System()
        system.add_walker(walker, origin)
        walk_simulation = walking(system, walker, distance)

        distances.append(round(walk_simulation, 1))

    return distances