peloton_main.jl

using Plots 

model = Model(50)

@gif for t ∈ 1:150            
    step!(model)
    
    postions = [ agent.position for agent in model.peloton ]
    arr = hcat(postions...)
    
    energy = [ agent.energy for agent in model.peloton ]   
    scatter(arr[1,:], arr[2,:], xlims=(-0.2, 1.), ylims=(0,1), marker_z=energy, color=:RdYlGn_9, clims=(0., 1), clabel="Energy")
end