corona.py

import matplotlib.pyplot as plt

p = 1000

i = 1

r = 3

imun = 0

def empfaenglich():
    return p - i - i

li = []
limun = []

while imun < 1000:
    neu = i * r
    print("Infiziert: ", i, "Neu:",neu,"Imun:",imun)
    li.append(i)
    limun.append(imun)
    imun += i
    i = neu
    
plt.interactive(True)

fig, ax = plt.subplots() 
ax.plot(range(1,len(li) + 1), li)
ax.plot(range(1,len(li) + 1), limun)

plt.savefig("graph.svg")