# Live programmering: populasjonsvekst

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

# Les fra kommandolinjen
x0 = float(sys.argv[1]) # Startpopulasjon
m = float(sys.argv[2])  # Max antall dyr
a = float(sys.argv[3])  # Vekstrate

# Opprett array
N = 200
x = np.zeros(N+1)
x[0] = x0

# Regn ut populasjon etter 1,2,3,... ?r
for n in range(1, N+1):
  val = x[n-1] + a * x[n-1] * (1 - x[n-1]/m)
  x[n] = round(val)

# Plott xn mot n
plt.plot(range(N+1), x, "r-")
plt.xlabel("?r")
plt.ylabel("Antall dyr")
plt.title("Logistisk vekst")
plt.show()

"""
Terminal> python populasjon.py 100 1000 0.05
"""