as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
def dySIS(y,t,lamda,mu):#SI/SIS...odeint数值解,求解微分方程初值问题
ySI=odeint(dySIS,i0,t,args=(lamda,0))#SI模型
ySIS=odeint(dySIS,i0,t,args=(lamda,mu))#SIS...模型
ySIR=odeint(dySIR,Y0,t,args=(lamda,mu))#SIR模型
#绘图
plt.title("SI、SIS和SIR模型的比较")
plt.xlabel('t-youcans...plt.axhline(y=0,ls="--",c='c')#添加水平直线
plt.plot(t,ySI,':g',label='i(t)-SI')
plt.plot(t,ySIS,'g--',label='i(t)-SIS