Модель SIDARTHE для прогнозирования эволюции covid-19 в Руанде с использованием ноутбука Jupyter - PullRequest
Новая
       26

Модель SIDARTHE для прогнозирования эволюции covid-19 в Руанде с использованием ноутбука Jupyter

0 голосов
/ 05 мая 2020

Я пытаюсь работать с моделью SIDARTHE для прогнозирования эволюции covid-19 в Руанде, но я застрял, мой код не может работать, я получаю ошибку значения. установка элемента массива с последовательностью, ниже мой код

# УПРАЖНЕНИЕ НА COVID 19 wanda 2020 #

## Импортируйте библиотеки ##

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint

Общая численность населения Руанды составляет 12 881 262 ###

Параметры

a=0.5700
b=0.0114
g=0.4560
d=0.0114
d=0.0114
q=0.3705
z=0.1254
h=0.1254
μ=0.0171
n=0.0274
w=0.0100
l=0.0342
r=0.0342
k=0.0171
x=0.0171
s=0.0171
#step=0.01;
#Initalization
y0=[1,3.3e-6,3.3e-7,1.6e-8,3.3e-8,0.0,0.0,0.0,]
t=np.linspace(0,60,20)
#t=np.arange(0,60)
#fucnction
def sidarthe(y,t,b,a,g,d,e,q,z,h,μ,n,w,l,r,k,x,s):
S,I,D,A,R,T,H,E=y 

   `dS=-S*a*I+b*D+g*A+s*R `  

     `dI=S*a*I+b*D+g*A+s*R-e*I-z*I-l*I`  

   `dD=e*I-h*D-r*D`  

    `dA=z*I-q*A-μ*A-k*A`  

    `dR=h*D+q*A-n*R-z*R`  

     `dT=μ*A+n*R-s*T-w*T`  

    `dH=l*I+r*D+k*A+z*R+s*T `  

    `dE=w*T`  

   `dndt=[dS,dI,dD,dA,dR,dT,dH,dE]`  

     `return dndt`

#solution
sol=odeint(sidarthe,y0,t,args=(b,a,g,d,e,q,z,h,μ,n,t,l,r,k,x,s))
#ploting
#print(t, sol[:,0])
plt.figure(figsize=(13,5))

plt.plot(t,sol[:,0],"b",label="Susceptible") 

`plt.plot(t,sol[:,1],"g",label="INFECTED") `

plt.plot(t,sol[:,2],"y",label="DIAGNOSED")

plt.plot(t,sol[:,3],"r",label="AILING")

plt.plot(t,sol[:,4],"o",label="RECOGNIZED")

plt.plot(t,sol[:,5],"p",label="THREATENED")

plt.plot(t,sol[:,6],"g--",label="HEALED")

plt.plot(t,sol[:,7],"r--",label="EXTINCT")
plt.legend(loc="best")

plt.xlabel("time(days)")

plt.ylabel("Fraction Population") 

`plt.show()`  


`when i run it am geting an error(Value error, setting an array element with a sequence)`
...