INF120/smitte.py

34 lines
1.1 KiB
Python
Raw Normal View History

2023-10-23 19:45:15 +00:00
# -*- coding: utf-8 -*-
"""Created on 23.10"""
__author__ = "Trygve B. Nomeland"
__email__ = "trygve.borte.nomeland@nmbu.no"
""""
SIR modell av en pandemi:
S (Susceptible) folk som ikke er eller har vært syke og som kan smittes
I (Infectious) folk som er syke og som kan smitte andre
R (Recoverd) folk har vært syke og har blitt friske igjen. Modellen antar at disse ikke kan bli syke igjen.
_y betyr verdi fra forrige steg
"""
def updateSIR(s_y, i_y, r_y, b, k):
return {'s': s_y - b*s_y*i_y, 'i': i_y + b*s_y*i_y-k*i_y, 'r': r_y + k*i_y}
# Generator som kjører simulasjonen frem til "stop" verdien
def genSIR(s_0, i_0, r_0, stop, b=1/3, k=1/10,):
SIR = {'s': s_0, 'i': i_0, 'r': r_0}
yield SIR
n = 0
while n < stop:
n += 1
SIR = updateSIR(SIR.get('s'), SIR.get('i'), SIR.get('r'), b, k)
yield SIR
def main():
f = open('pan.csv', 'w')
f.write('Day,s@day,i@day,r@day\n')
for n, S in enumerate(genSIR(1, 10/(5*10**6), 0, stop=120)):
f.write(f'{n},{S.get('s'):.3e},{S.get('i'):.3e},{S.get('r'):.3e}\n')
f.close()
if __name__ == '__main__':
main()